burn and sb3-burn
`sb3-burn` is an ecosystem sibling of `burn`, serving as an implementation of stable-baselines3 that leverages the `burn` tensor library and Deep Learning Framework.
About burn
tracel-ai/burn
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
Leverages Rust's compile-time guarantees to enable static-graph optimizations while maintaining dynamic flexibility, with a composable backend architecture supporting GPU (CUDA, ROCm, Metal, Vulkan, WebGPU) and CPU execution. Backend decorators like Autodiff and Fusion augment any backend with automatic differentiation and kernel fusion without code changes. Supports distributed training via Remote backend and seamless CPU-GPU operation routing, while unifying training and inference workflows with no code modifications needed between them.
About sb3-burn
will-maclean/sb3-burn
Implementation of stable-baselines3 in rust with burn
Scores updated daily from GitHub, PyPI, and npm data. How scores work