will-maclean / sb3-burnLinks
Implementation of stable-baselines3 in rust with burn
☆15Updated 8 months ago
Alternatives and similar repositories for sb3-burn
Users that are interested in sb3-burn are comparing it to the libraries listed below
Sorting:
- RAI: Rust ML framework with composable transformations like JAX.☆89Updated 10 months ago
- OpenAI's Gym written in pure Rust for blazingly fast performance☆99Updated 7 months ago
- JAX implementation of RL algorithms and vectorized environments☆45Updated last year
- Reinforcement learning training framework for entity-gym environments.☆17Updated last year
- Various reinforcement learning algorithms written in Jax + Flax☆24Updated last year
- On-the-fly conversions between Jax and NumPy tensors☆51Updated 2 years ago
- Standard interface for entity based reinforcement learning environments.☆37Updated last year
- JAX implementations of core Deep RL algorithms☆79Updated 3 years ago
- Efficient numpy-like ragged array datatype for Python.☆19Updated 2 years ago
- a modular reinforcement learning library with JAX agents☆26Updated 3 months ago
- Baselines for gymnax 🤖☆66Updated 2 years ago
- Fast QP Solver in JAX☆13Updated 9 months ago
- A reinforcement learning library in Rust☆53Updated 3 weeks ago
- JAxtar is a project with a JAX-native implementation of parallelizeable A* & Q* solver for neural heuristic search research.☆24Updated this week
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆66Updated last year
- A PyTorch library for all things nonlinear control and reinforcement learning.☆46Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆56Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆73Updated 9 months ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆36Updated 2 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆22Updated 4 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆22Updated 3 months ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆53Updated 3 weeks ago
- ☆213Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆48Updated this week
- Mathematical Programming in JAX☆83Updated this week
- Belief-state planning for POMDPs using learned approximations☆21Updated 4 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆31Updated last month
- Fully differentiable RL environments, written in Ivy.☆65Updated last year
- The core repository of the elsciRL framework.☆16Updated this week