flowersteam / vivarium
Multi-agent simulator in Jax for research and teaching in AI & ALife
☆15Updated last week
Alternatives and similar repositories for vivarium:
Users that are interested in vivarium are comparing it to the libraries listed below
- ☆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
- Reinforcement Learning inside a 3D soccer simulation☆25Updated 5 months ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆13Updated 8 months ago
- ☆20Updated 8 months ago
- POPGym Library in JAX☆11Updated 10 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆13Updated last month
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆17Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Simple JAX Graphics Library.☆30Updated 3 months ago
- ☆18Updated 3 weeks ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆13Updated last month
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 7 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆48Updated last year
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆13Updated last year
- a modular reinforcement learning library with JAX agents☆22Updated last year
- General Modules for JAX☆63Updated 6 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆37Updated last year
- ☆71Updated 6 months ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆24Updated 7 months ago
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago