araffin / sbxLinks
SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
ā523Updated 3 weeks ago
Alternatives and similar repositories for sbx
Users that are interested in sbx are comparing it to the libraries listed below
Sorting:
- RL Environments in JAX šā826Updated 5 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesā453Updated 3 weeks ago
- šļøA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX ⢠End-to-End JAX RLā368Updated 2 weeks ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.ā728Updated 3 years ago
- Really Fast End-to-End Jax RL Implementationsā969Updated last year
- Multi-Agent Reinforcement Learning with JAXā679Updated this week
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!ā244Updated 2 weeks ago
- ā241Updated 11 months ago
- Distributed Reinforcement Learning accelerated by Lightning Fabricā405Updated last week
- Multi-objective Gymnasium environments for reinforcement learningā351Updated 4 months ago
- Partially Observable Process Gymā204Updated 5 months ago
- An API conversion tool for popular external reinforcement learning environmentsā191Updated 2 weeks ago
- A framework for Reinforcement Learning research.ā171Updated this week
- ā” Flashbax: Accelerated Replay Buffers in JAXā258Updated last month
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.ā317Updated last year
- A Simplified Pytorch Version of the Dreamer Algorithmā140Updated 2 years ago
- Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lā¦ā897Updated 3 weeks ago
- A Python interface for reinforcement learning environmentsā385Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imaginationā562Updated 4 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithmā233Updated 2 years ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022ā339Updated last year
- š¹ļø A diverse suite of scalable reinforcement learning environments in JAXā771Updated last week
- Simple single-file baselines for Q-Learning in pure-GPU settingā188Updated 7 months ago
- High throughput synchronous and asynchronous reinforcement learningā949Updated last week
- Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) codeā657Updated last week
- Baseline implementation of recurrent PPO using truncated BPTTā155Updated last year
- ā355Updated 3 years ago
- Reinforcement learning algorithms for MuJoCo tasksā430Updated 8 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.ā852Updated last year
- A collection of robotics simulation environments for reinforcement learningā796Updated 2 weeks ago