instadeepai / flashbax
⥠Flashbax: Accelerated Replay Buffers in JAX
â228Updated this week
Alternatives and similar repositories for flashbax:
Users that are interested in flashbax are comparing it to the libraries listed below
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ300Updated this week
- â191Updated 3 months ago
- Accelerated minigrid environments with JAXâ132Updated 7 months ago
- â73Updated 4 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ149Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â294Updated last month
- Accelerated Quality-Diversityâ280Updated this week
- A neural network library written in jaxâ11Updated last month
- Clean single-file implementation of offline RL algorithms in JAXâ137Updated 3 months ago
- Baselines for gymnax đ¤â66Updated last year
- RL Environments in JAX đâ722Updated 8 months ago
- A tool for aggregating and plotting MARL experiment data.â76Updated 2 months ago
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ57Updated last year
- JAX implementations of core Deep RL algorithmsâ79Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ109Updated 7 months ago
- Modular framework for Reinforcement Learning in pythonâ172Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ423Updated 2 weeks ago
- â74Updated this week
- Multi-Agent Reinforcement Learning with JAXâ541Updated 2 weeks ago
- Efficient baselines for autocurricula in JAX.â186Updated 6 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ705Updated this week
- Benchmarking RL generalization in an interpretable way.â151Updated 2 weeks ago
- â217Updated 4 months ago
- Partially Observable Process Gymâ183Updated 8 months ago
- JAX implementation of RL algorithms and vectorized environmentsâ42Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]â99Updated last year
- Goal-Conditioned Reinforcement Learning with JAXâ131Updated this week
- Evaluating long-term memory of reinforcement learning algorithmsâ141Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learningâ71Updated 7 months ago
- â298Updated 3 months ago