gitwtfhub

wtf is adajepa?

agentic-learning-ai-lab/adajepa — explained in plain English

Analysis updated 2026-05-18

112PythonAudience · researcherComplexity · 5/5Setup · hard

TL;DR

Research code for AdaJEPA, a world model that keeps adapting itself while planning actions, instead of staying frozen.

Mindmap

mindmap
  root((AdaJEPA))
    What it does
      Adaptive world model
      Plans then updates
      Closed loop MPC
    Evaluations
      Shape shift
      Visual shift
      Dynamics shift
      Layout shift
    Requirements
      Conda environment
      Pretrained checkpoints
      MuJoCo for maze
    Builds on
      Temporal straightening
      DINO-WM

Code map

Detail Auto

An interactive map of this repo's files and how they connect — its source is parsed live in your browser. Click Visualize to build it.

filefunction / class

Why would anyone build with this?

REASON 1

Reproduce the AdaJEPA paper's planning results on the released pushing and maze tasks.

REASON 2

Study how test-time model adaptation improves planning under distribution shift.

REASON 3

Evaluate a pretrained world model against visual, dynamics, or layout changes.

REASON 4

Compare AdaJEPA's adaptive planner against the frozen MPC baseline included in the code.

What's in the stack?

PythonPyTorchMuJoCoConda

How it stacks up

agentic-learning-ai-lab/adajepacanwhite/microragexploitbench/exploitbench
Stars112112112
LanguagePythonPythonPython
Setup difficultyhardeasyhard
Complexity5/54/5
Audienceresearcherresearcherresearcher

Figures from each repo's GitHub metadata at analysis time.

How do you spin it up?

Difficulty · hard Time to first run · 1day+

Requires a conda environment, separately downloaded checkpoints and data, and MuJoCo for some maze evaluations.

Wtf does this do

This is the official code release for a research paper called AdaJEPA, which is about how an AI system can plan actions in the real world, such as pushing an object to a target spot or navigating a maze. Systems like this often use a world model, a learned representation that predicts what the next state of the environment will look like after taking an action, and then plans a sequence of actions by imagining several possible futures. Normally that world model is trained once and then kept frozen while it is being used, so if conditions change from what it was trained on, its predictions get worse and planning can fail. AdaJEPA's contribution is to keep adjusting the world model while it is being used, instead of leaving it frozen. After planning and taking its first action, the system compares what it predicted would happen to what actually happened, uses that difference as a signal to update the model slightly, and then plans again with the improved model before taking the next action. This update loop repeats continuously as the task proceeds, without needing any extra human demonstrations, and the paper reports that this substantially improves planning success even with just one small update step between each replanning. The code builds on two earlier research projects, temporal-straightening and DINO-WM, and is meant for other researchers working in this area rather than for direct product use. It provides the training and adaptation code, configuration files for several types of environment shifts such as visual corruption, changed object shapes, changed physics, and unfamiliar maze layouts, and links to download pretrained checkpoints and evaluation data. Setup involves cloning the repository, creating the provided conda environment, and downloading separately hosted checkpoint and data files. Certain maze evaluations additionally require the MuJoCo physics simulator. The README includes example commands for running each type of evaluation and for visualizing what the model imagines compared to the real environment and the goal. The project includes a citation entry for the associated academic paper. No license file is described in the excerpt provided.

Yoink these prompts

Prompt 1
Walk me through setting up the conda environment and downloading the checkpoints for AdaJEPA.
Prompt 2
Explain in plain terms how AdaJEPA adapts its world model during planning instead of staying frozen.
Prompt 3
Show me how to run the visual shift evaluation on the pusht_visual_shift checkpoint.
Prompt 4
Explain the difference between the frozen MPC baseline and the AdaJEPA planner in this repo.

Frequently asked questions

wtf is adajepa?

Research code for AdaJEPA, a world model that keeps adapting itself while planning actions, instead of staying frozen.

What language is adajepa written in?

Mainly Python. The stack also includes Python, PyTorch, MuJoCo.

How hard is adajepa to set up?

Setup difficulty is rated hard, with roughly 1day+ to a first successful run.

Who is adajepa for?

Mainly researcher.

View the repo → Decode another repo

This repo across BitVibe Labs

Don't trust strangers blindly. Verify against the repo.