arnavkj1995 / SFM
Official Implementation of SFM and the baselines in Jax.
☆15Updated 2 months ago
Alternatives and similar repositories for SFM:
Users that are interested in SFM are comparing it to the libraries listed below
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆13Updated 7 months ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆15Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Reinforcement Learning inside a 3D soccer simulation☆25Updated 4 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆54Updated 10 months ago
- ☆42Updated 6 months ago
- ☆34Updated last month
- Scalable Opponent Shaping Experiments in JAX☆24Updated 9 months ago
- ☆15Updated 2 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆25Updated 6 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- ☆67Updated 5 months ago
- PyTorch Package For Quasimetric Learning☆41Updated 2 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆15Updated 6 months ago
- ☆13Updated 7 months ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆27Updated last year
- ☆29Updated last year
- ☆26Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆26Updated 3 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆15Updated last week
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆61Updated 10 months ago
- ☆31Updated 10 months ago
- My Body Is A Cage☆38Updated 3 years ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 9 months ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆24Updated 7 months ago
- ☆13Updated 6 months ago