thrust-rl 0.4.0

High-performance reinforcement learning in Rust with the Burn tensor backend
Documentation
1
2
3
4
5
6
7
8
9
10
11
//! Experience buffers and replay management
//!
//! This module handles storage and sampling of experience for training.

pub mod replay;
pub mod rollout;

// Convenience re-export so multi-agent training scripts can pull the
// flat-buffer GAE helper directly off `thrust_rl::buffer`. Single-agent
// users continue to call `RolloutBuffer::compute_advantages`.
pub use rollout::compute_advantages_multi_agent;