Eclectic-Sheep / sheeprl
Distributed Reinforcement Learning accelerated by Lightning Fabric
β355Updated this week
Alternatives and similar repositories for sheeprl:
Users that are interested in sheeprl are comparing it to the libraries listed below
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsβ423Updated 2 weeks ago
- ποΈA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX β’ End-to-End JAX RLβ300Updated this week
- RL Environments in JAX πβ722Updated 8 months ago
- Implementation of Dreamer v3 in pytorch.β511Updated 5 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesβ362Updated this week
- Multi-Agent Reinforcement Learning with JAXβ541Updated 2 weeks ago
- Multi-objective Gymnasium environments for reinforcement learningβ311Updated 2 weeks ago
- An API conversion tool for popular external reinforcement learning environmentsβ156Updated 2 months ago
- β217Updated 4 months ago
- β191Updated 3 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.β668Updated 2 years ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ228Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.β294Updated last month
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.β524Updated 4 months ago
- A collection of MARL benchmarks based on TorchRLβ366Updated this week
- Really Fast End-to-End Jax RL Implementationsβ841Updated 6 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ149Updated this week
- High throughput synchronous and asynchronous reinforcement learningβ882Updated 2 months ago
- Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in Openβ¦β264Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.β297Updated last year
- Accelerated minigrid environments with JAXβ132Updated 7 months ago
- A Simplified Pytorch Version of the Dreamer Algorithmβ123Updated last year
- Deep reinforcement learning without experience replay, target networks, or batch updates.β225Updated this week
- Partially Observable Process Gymβ183Updated 8 months ago
- PyTorch implementation of DreamerV2 model-based RL algorithmβ216Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.β813Updated 7 months ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.β698Updated this week
- β338Updated 2 years ago
- A collection of robotics simulation environments for reinforcement learningβ643Updated 2 months ago
- DrQ-v2: Improved Data-Augmented Reinforcement Learningβ379Updated 2 years ago