LICENSE
README.md
setup.py
jepa_src/__init__.py
jepa_src/datasets/__init__.py
jepa_src/datasets/data_manager.py
jepa_src/datasets/image_dataset.py
jepa_src/datasets/video_dataset.py
jepa_src/datasets/utils/__init__.py
jepa_src/datasets/utils/weighted_sampler.py
jepa_src/datasets/utils/video/__init__.py
jepa_src/datasets/utils/video/functional.py
jepa_src/datasets/utils/video/randaugment.py
jepa_src/datasets/utils/video/randerase.py
jepa_src/datasets/utils/video/transforms.py
jepa_src/datasets/utils/video/volume_transforms.py
jepa_src/masks/__init__.py
jepa_src/masks/default.py
jepa_src/masks/multiblock3d.py
jepa_src/masks/random_tube.py
jepa_src/masks/utils.py
jepa_src/models/__init__.py
jepa_src/models/attentive_pooler.py
jepa_src/models/predictor.py
jepa_src/models/vision_transformer.py
jepa_src/models/utils/__init__.py
jepa_src/models/utils/modules.py
jepa_src/models/utils/multimask.py
jepa_src/models/utils/patch_embed.py
jepa_src/models/utils/pos_embs.py
jepa_src/utils/__init__.py
jepa_src/utils/distributed.py
jepa_src/utils/functional.py
jepa_src/utils/logging.py
jepa_src/utils/monitoring.py
jepa_src/utils/schedulers.py
jepa_src/utils/tensors.py
vjepa_encoder/__init__.py
vjepa_encoder/vision_encoder.py
vjepa_encoder.egg-info/PKG-INFO
vjepa_encoder.egg-info/SOURCES.txt
vjepa_encoder.egg-info/dependency_links.txt
vjepa_encoder.egg-info/requires.txt
vjepa_encoder.egg-info/top_level.txt
vjepa_encoder/vjepa/__init__.py
vjepa_encoder/vjepa/train.py
vjepa_encoder/vjepa/transforms.py
vjepa_encoder/vjepa/utils.py