Train a JEPA world model on a set of pre-collected trajectories from a toy environment involving an agent in two rooms.
deep-learning recurrent-networks linear-probing curriculum-learning energy-based-model self-supervised-learning spatial-embeddings vicreg jepa world-model joint-embedding-prediction-architecture agent-trajectory latent-prediction
-
Updated
Dec 17, 2024 - Python