araffin / sbxLinks
SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
ā512Updated last week
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 šā817Updated 4 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesā448Updated 2 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.ā723Updated 2 years ago
- šļøA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX ⢠End-to-End JAX RLā362Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementationsā963Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!ā238Updated 4 months ago
- Multi-objective Gymnasium environments for reinforcement learningā346Updated 3 months ago
- ā239Updated 11 months ago
- Partially Observable Process Gymā202Updated 4 months ago
- An API conversion tool for popular external reinforcement learning environmentsā191Updated last month
- A framework for Reinforcement Learning research.ā163Updated last month
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.ā313Updated last year
- ā” Flashbax: Accelerated Replay Buffers in JAXā255Updated last month
- Distributed Reinforcement Learning accelerated by Lightning Fabricā398Updated last week
- A Python interface for reinforcement learning environmentsā379Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imaginationā557Updated 4 years ago
- Baseline implementation of recurrent PPO using truncated BPTTā155Updated last year
- Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) codeā648Updated last month
- PyTorch implementation of DreamerV2 model-based RL algorithmā229Updated 2 years ago
- ā287Updated 3 years ago
- Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lā¦ā890Updated this week
- A Simplified Pytorch Version of the Dreamer Algorithmā139Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.ā166Updated this week
- DrQ-v2: Improved Data-Augmented Reinforcement Learningā409Updated 3 years ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022ā335Updated last year
- Imitation learning algorithmsā553Updated 7 months ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.ā190Updated 2 weeks ago
- A collection of robotics simulation environments for reinforcement learningā785Updated last month
- Accelerated Quality-Diversityā324Updated 4 months ago
- Reinforcement learning algorithms for MuJoCo tasksā424Updated 7 months ago