instadeepai / flashbax
⥠Flashbax: Accelerated Replay Buffers in JAX
â227Updated 2 weeks ago
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â287Updated this week
- â191Updated 3 months ago
- Accelerated minigrid environments with JAXâ132Updated 7 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â290Updated 2 weeks ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ141Updated 3 months ago
- â73Updated 4 months ago
- A neural network library written in jaxâ11Updated last month
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ57Updated last year
- Accelerated Quality-Diversityâ279Updated this week
- A tool for aggregating and plotting MARL experiment data.â73Updated last month
- â74Updated 6 months ago
- Baselines for gymnax đ¤â66Updated last year
- Partially Observable Process Gymâ181Updated 8 months ago
- Clean single-file implementation of offline RL algorithms in JAXâ134Updated 2 months ago
- JAX implementation of RL algorithms and vectorized environmentsâ41Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ109Updated 6 months ago
- Benchmarking RL generalization in an interpretable way.â147Updated this week
- RL Environments in JAX đâ717Updated 8 months ago
- Modular framework for Reinforcement Learning in pythonâ172Updated 2 years ago
- â217Updated 3 months ago
- Goal-Conditioned Reinforcement Learning with JAXâ127Updated this week
- Evaluating long-term memory of reinforcement learning algorithmsâ140Updated last year
- JAX implementations of core Deep RL algorithmsâ79Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAXâ531Updated last week
- Efficient baselines for autocurricula in JAX.â185Updated 6 months ago
- â297Updated 2 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ416Updated this week
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ701Updated last month
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ66Updated 2 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]â97Updated last year