Skip to content

benbaarber/rl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

rl - A rust reinforcement learning library

Current Crates.io Version Documentation Rust Version

NOTE: Progress will be slow until later this month when I finish studying Barto & Sutton's RL textbook to gain a stronger theoretical understanding of the field

About

rl is a fully Rust-native reinforcement learning library with the goal of providing a unified RL development experience, aiming to do for RL what libraries like PyTorch did for deep learning. By leveraging Rust's powerful type system and the burn library, rl enables users to reuse production-ready SoTA algorithms with arbitrary environments, state spaces, and action spaces.

This project also aims to provide a clean platform for experimentation with new RL algorithms. By combining burn's powerful deep learning features with rl's provided RL sub-algorithms and components, users can create, test, and benchmark their own new experimental agents without having to start from scratch.

Currently, rl is in its early stages. Contributors are more than welcome!

Features

  • High-performance production-ready implementations of all SoTA RL algorithms
  • Detailed logging and training visualization TUI (see image below)
  • Maximum extensibility for creating and testing new experimental algorithms
  • Gym environments
  • A comfortable learning experience for those new to RL
  • General RL peripherals and utility functions

TUI example