MCPcopy Index your code
hub / github.com/benbaarber/rl

github.com/benbaarber/rl @v0.4.0

Chat with this repo
repository ↗ · DeepWiki ↗ · release v0.4.0 ↗ · + Follow
200 symbols 371 edges 35 files 66 documented · 33%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

rl - A rust reinforcement learning library

Current Crates.io Version

About

rl is a fully rust-native reinforcement learning library with the goal of providing a unified RL development experience, including: - High-performance production-ready implementations of all SoTA RL algorithms powered by the rust-native deep learning framework burn - Out of the box benchmarking multiple algorithms against eachother - Detailed logging and training visualization - 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

Extension points exported contracts — how you extend this code

Decay (Interface)
An implementation of a time-decaying value [5 implementers]
src/decay.rs
Environment (Interface)
Represents a Markov decision process, defining the dynamics of an environment in which an agent can operate. This base [4 …
src/env.rs
DQNModel (Interface)
A burn module used with a Deep Q network agent ### Generics - `B`: A burn backend - `D`: The dimension of the input ten [1 …
src/algo/dqn.rs
ProbModel (Interface)
Trait for probabilistic models ### Type parameters - `T`: The type returned by sampling the distribution - `O`: The typ
src/prob/mod.rs
Component (Interface)
(no doc) [2 implementers]
src/viz/components/mod.rs
DiscreteActionSpace (Interface)
An [Environment] with a discrete action space [3 implementers]
src/env.rs
ToTensor (Interface)
A trait for converting items to tensors Commonly implemented for `Vec ` to convert batches of `T` to a tensor of dime [2 …
src/env.rs

Core symbols most depended-on inside this repo

push
called by 13
src/memory/base.rs
len
called by 12
src/viz/components/plot.rs
render
called by 12
src/viz/components/heatmap_scatter_plot/mod.rs
clone
called by 10
src/memory/exp.rs
take
called by 7
src/env.rs
choose
called by 7
src/exploration/ucb.rs
block
called by 6
src/viz/components/heatmap_scatter_plot/mod.rs
set_style
called by 6
src/viz/components/heatmap_scatter_plot/mod.rs

Shape

Method 119
Class 37
Function 31
Interface 7
Enum 6

Languages

Rust100%

Modules by API surface

src/viz/components/heatmap_scatter_plot/mod.rs30 symbols
src/gym/grassy_field.rs21 symbols
src/env.rs15 symbols
src/decay.rs15 symbols
src/viz/components/plot.rs11 symbols
src/gym/frozen_lake.rs10 symbols
src/gym/cart_pole.rs10 symbols
src/ds.rs10 symbols
src/algo/q_table.rs8 symbols
src/algo/dqn.rs8 symbols
src/viz/app.rs7 symbols
src/memory/base.rs7 symbols

For agents

$ claude mcp add rl \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact