instadeepai / jumanjiLinks
đšī¸ A diverse suite of scalable reinforcement learning environments in JAX
â737Updated this week
Alternatives and similar repositories for jumanji
Users that are interested in jumanji are comparing it to the libraries listed below
Sorting:
- RL Environments in JAX đâ771Updated 3 weeks ago
- Accelerated Quality-Diversityâ304Updated last week
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ335Updated last week
- Multi-Agent Reinforcement Learning with JAXâ595Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementationsâ898Updated 9 months ago
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ808Updated this week
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ238Updated 2 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ467Updated this week
- Evolution Strategies in JAX đĻâ636Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â685Updated 2 years ago
- A suite of test scenarios for multi-agent reinforcement learning.â711Updated 3 weeks ago
- âī¸ Vectorized RL game environments in JAXâ487Updated 3 months ago
- Distributed Reinforcement Learning accelerated by Lightning Fabricâ380Updated 2 weeks ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â226Updated 3 weeks ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â833Updated 10 months ago
- Multi-objective Gymnasium environments for reinforcement learningâ331Updated 3 months ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ67Updated 2 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â317Updated 3 weeks ago
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Researchâ493Updated 4 months ago
- A collection of MARL benchmarks based on TorchRLâ414Updated last month
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesâ398Updated 2 weeks ago
- â900Updated 11 months ago
- VMAS is a vectorized differentiable simulator designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of âĻâ433Updated 2 weeks ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)â486Updated last month
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.â597Updated 7 months ago
- â1,329Updated last month
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.â783Updated this week
- An engine for high performance multi-agent environments with very large numbers of agents, along with a set of reference environmentsâ295Updated 4 months ago
- A Python interface for reinforcement learning environmentsâ368Updated 2 years ago
- â232Updated 7 months ago