keraJLi / synthetic-gymnax
☆20Updated 8 months ago
Alternatives and similar repositories for synthetic-gymnax:
Users that are interested in synthetic-gymnax are comparing it to the libraries listed below
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- ☆13Updated 7 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 3 months ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated 10 months ago
- Simple JAX Graphics Library.☆30Updated 3 months ago
- ☆70Updated 5 months ago
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- Learning diverse options through the Laplacian representation.☆23Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆95Updated last year
- ☆18Updated 3 weeks ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆13Updated 8 months ago
- POPGym Library in JAX☆11Updated 10 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆24Updated 7 months ago
- Reinforcement Learning inside a 3D soccer simulation☆25Updated 5 months ago
- ☆31Updated 11 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 6 months ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆13Updated last year
- General Modules for JAX☆63Updated 6 months ago
- Corax: Core RL in JAX☆36Updated 11 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆15Updated last week
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆48Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆55Updated 10 months ago
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…☆27Updated 2 years ago
- Baselines for gymnax 🤖☆63Updated last year
- ☆73Updated 3 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆20Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆137Updated 2 months ago