LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
docs/build_docs.py
docs/requirements.txt
docs/archive/REUSABLE_JEPA.md
docs/archive/STRUCTURED_DATA.md
docs/source/conf-original.py
docs/source/conf.py
docs/source/index.md
docs/source/_static/robots.txt
docs/source/api/cli.md
docs/source/api/config.md
docs/source/api/data.md
docs/source/api/logging.md
docs/source/api/models.md
docs/source/api/training.md
docs/source/concepts/architecture.md
docs/source/concepts/self_supervised_learning.md
docs/source/concepts/structured_data.md
docs/source/examples/index.md
docs/source/guides/configuration.md
docs/source/guides/data.md
docs/source/guides/installation.md
docs/source/guides/quickstart.md
docs/source/guides/training.md
examples/cli_example.py
examples/data_example.py
examples/hf_example.py
examples/logging_example.py
examples/structured_data_example.py
examples/training_example.py
examples/usage_example.py
examples/wandb_example.py
jepa/__init__.py
jepa/py.typed
jepa.egg-info/PKG-INFO
jepa.egg-info/SOURCES.txt
jepa.egg-info/dependency_links.txt
jepa.egg-info/entry_points.txt
jepa.egg-info/not-zip-safe
jepa.egg-info/requires.txt
jepa.egg-info/top_level.txt
jepa/cli/__init__.py
jepa/cli/__main__.py
jepa/cli/evaluate.py
jepa/cli/train.py
jepa/cli/utils.py
jepa/config/__init__.py
jepa/config/config.py
jepa/config/default_config.yaml
jepa/config/nlp_config.yaml
jepa/config/timeseries_config.yaml
jepa/config/vision_config.yaml
jepa/data/__init__.py
jepa/data/dataset.py
jepa/data/hf_compatibility.py
jepa/data/transforms.py
jepa/data/utils.py
jepa/loggers/__init__.py
jepa/loggers/base_logger.py
jepa/loggers/console_logger.py
jepa/loggers/multi_logger.py
jepa/loggers/tensorboard_logger.py
jepa/loggers/wandb_logger.py
jepa/loss_functions/__init__.py
jepa/loss_functions/loss_functions.py
jepa/models/__init__.py
jepa/models/base.py
jepa/models/encoder.py
jepa/models/jepa.py
jepa/models/predictor.py
jepa/trainer/__init__.py
jepa/trainer/eval.py
jepa/trainer/trainer.py
jepa/trainer/utils.py
tests/__init__.py
tests/test_basic.py
tests/test_cli.py
tests/test_config.py
tests/test_config_data.py
tests/test_data.py
tests/test_logging.py
tests/test_model.py
tests/test_trainer.py