will-maclean/sb3-burn
Implementation of stable-baselines3 in rust with burn
This is a library for machine learning engineers who need to build reinforcement learning systems in Rust. It takes descriptions of environments and desired behaviors, then produces trained AI agents that can learn to perform tasks through trial and error. The library offers common reinforcement learning algorithms and environments, with a focus on speed and flexibility for custom scenarios.
Use this if you are a Rust developer working on AI agents and need a fast, extensible framework for reinforcement learning, especially if you require GPU support beyond Nvidia.
Not ideal if you are a Python developer or prefer using established Python-based reinforcement learning libraries like Stable-baselines3.
Stars
20
Forks
1
Language
Rust
License
MIT
Category
Last pushed
Nov 24, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/will-maclean/sb3-burn"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Compare
Higher-rated alternatives
tracel-ai/burn
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on...
sonos/tract
Tiny, no-nonsense, self-contained, Tensorflow and ONNX inference
pykeio/ort
Fast ML inference & training for ONNX models in Rust
elixir-nx/ortex
ONNX Runtime bindings for Elixir
robertknight/rten
ONNX neural network inference engine