LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
configs/ijepa_base.yaml
configs/lewm_base.yaml
configs/vjepa_base.yaml
pyjepa/__init__.py
pyjepa/_version.py
pyjepa/cli.py
pyjepa.egg-info/PKG-INFO
pyjepa.egg-info/SOURCES.txt
pyjepa.egg-info/dependency_links.txt
pyjepa.egg-info/entry_points.txt
pyjepa.egg-info/requires.txt
pyjepa.egg-info/top_level.txt
pyjepa/data/__init__.py
pyjepa/data/image_dataset.py
pyjepa/data/normalize.py
pyjepa/data/trajectory.py
pyjepa/data/transforms.py
pyjepa/data/video_dataset.py
pyjepa/device/__init__.py
pyjepa/device/backend.py
pyjepa/device/precision.py
pyjepa/inference/__init__.py
pyjepa/inference/encoder.py
pyjepa/inference/rollout.py
pyjepa/inference/surprise.py
pyjepa/losses/__init__.py
pyjepa/losses/jepa.py
pyjepa/losses/prediction.py
pyjepa/losses/sigreg.py
pyjepa/losses/vicreg.py
pyjepa/masks/__init__.py
pyjepa/masks/multiblock.py
pyjepa/masks/random_patch.py
pyjepa/masks/tube.py
pyjepa/masks/utils.py
pyjepa/models/__init__.py
pyjepa/models/action_encoder.py
pyjepa/models/ijepa.py
pyjepa/models/lewm.py
pyjepa/models/predictor.py
pyjepa/models/registry.py
pyjepa/models/vit.py
pyjepa/models/vjepa.py
pyjepa/models/wrappers.py
pyjepa/nn/__init__.py
pyjepa/nn/attention.py
pyjepa/nn/mlp.py
pyjepa/nn/patch_embed.py
pyjepa/nn/pooler.py
pyjepa/nn/pos_embed.py
pyjepa/nn/transformer.py
pyjepa/optim/__init__.py
pyjepa/optim/ema.py
pyjepa/optim/factory.py
pyjepa/optim/schedulers.py
pyjepa/planning/__init__.py
pyjepa/planning/cem.py
pyjepa/planning/mppi.py
pyjepa/planning/policy.py
pyjepa/planning/random_shoot.py
pyjepa/training/__init__.py
pyjepa/training/base.py
pyjepa/training/finetune.py
pyjepa/training/ijepa.py
pyjepa/training/lewm.py
pyjepa/training/vjepa.py
pyjepa/utils/__init__.py
pyjepa/utils/checkpoint.py
pyjepa/utils/config.py
pyjepa/utils/distributed.py
pyjepa/utils/logging.py
pyjepa/utils/tensors.py
tests/test_device.py
tests/test_e2e.py
tests/test_losses.py
tests/test_masks.py
tests/test_models.py
tests/test_planning.py