arnavkj1995 / SFM
Official Implementation of SFM and the baselines in Jax.
☆15Updated 3 months ago
Alternatives and similar repositories for SFM:
Users that are interested in SFM are comparing it to the libraries listed below
- GPT implementation in Flax☆18Updated 3 years ago
- Reinforcement Learning inside a 3D soccer simulation☆25Updated 5 months ago
- Generalised UDRL☆37Updated 2 years ago
- ☆27Updated last year
- Corax: Core RL in JAX☆36Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆13Updated 8 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆54Updated 11 months ago
- ☆15Updated 2 years ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆24Updated 8 months ago
- ☆31Updated 11 months ago
- Action Value Gradient Algorithm☆18Updated 2 weeks ago
- ☆13Updated 8 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- ☆73Updated 6 months ago
- General Modules for JAX☆64Updated last week
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 7 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- My Body Is A Cage☆39Updated 3 years ago
- ☆41Updated 7 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 6 months ago
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆25Updated 2 years ago
- Fast reinforcement learning research☆57Updated 2 months ago
- ☆28Updated last year
- ☆18Updated 5 months ago
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year