flowersteam / vivarium
Multi-agent simulator in Jax for research and teaching in AI & ALife
☆27Updated this week
Alternatives and similar repositories for vivarium:
Users that are interested in vivarium are comparing it to the libraries listed below
- ☆77Updated last month
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆66Updated 8 months ago
- ☆39Updated 3 years ago
- POPGym Library in JAX☆11Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated 8 months ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- ☆13Updated 9 months ago
- ☆41Updated 9 months ago
- Reinforcement Learning inside a 3D soccer simulation☆26Updated 7 months ago
- flexible meta-learning in jax☆13Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆47Updated 4 months ago
- General Modules for JAX☆64Updated last month
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Updated 2 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Highly scalable 2D JAX physics engine.☆56Updated last month
- Action Value Gradient Algorithm☆20Updated last month
- ☆31Updated last year
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆10Updated last year
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- DreamerV3 implementation of Curious Replay, a method for prioritizing experience replay that is tailored to model-based reinforcement lea…☆36Updated last year
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆26Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆20Updated 10 months ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆15Updated last month
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 9 months ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆54Updated 2 years ago
- Simple JAX Graphics Library.☆36Updated 6 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆55Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆24Updated last year