ikostrikov / jaxrlLinks
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
â707Updated 2 years ago
Alternatives and similar repositories for jaxrl
Users that are interested in jaxrl are comparing it to the libraries listed below
Sorting:
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â836Updated last year
- RL Environments in JAX đâ796Updated 2 months ago
- Library for Model Based RLâ1,012Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ493Updated 2 weeks ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesâ424Updated this week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (âĻâ475Updated last year
- A Python interface for reinforcement learning environmentsâ372Updated 2 years ago
- Really Fast End-to-End Jax RL Implementationsâ939Updated 11 months ago
- Dream to Control: Learning Behaviors by Latent Imaginationâ548Updated 3 years ago
- Multi-Agent Reinforcement Learning with JAXâ624Updated last month
- â351Updated 2 years ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"â508Updated 2 years ago
- Code for conservative Q-learningâ452Updated 3 years ago
- Imitation learning algorithmsâ547Updated 5 months ago
- Multi-objective Gymnasium environments for reinforcement learningâ340Updated last month
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.â306Updated last year
- â310Updated 8 months ago
- â235Updated 9 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ754Updated 2 months ago
- PyTorch implementation of Soft Actor-Critic (SAC)â556Updated 3 years ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022â327Updated last year
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of codeâ700Updated last year
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 gamesâ553Updated 2 years ago
- High throughput synchronous and asynchronous reinforcement learningâ927Updated 3 months ago
- Code for the paper "Phasic Policy Gradient"â262Updated 2 years ago
- Tools for accelerating safe exploration research.â550Updated 2 years ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ345Updated 3 weeks ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RLâ368Updated 3 years ago
- Real-World RL Benchmark Suiteâ355Updated 5 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithmâ226Updated 2 years ago