rlox_candle/lib.rs
1//! rlox-candle: Candle backend for pure-Rust neural network inference and training.
2//!
3//! Implements [`rlox_nn::ActorCritic`] and other NN traits using
4//! [Candle](https://github.com/huggingface/candle) (HuggingFace's Rust ML framework).
5//!
6//! Key components:
7//! - [`actor_critic::CandleActorCritic`]: Discrete actor-critic with PPO step.
8//! - [`collector`]: `SharedPolicy` + `make_candle_callbacks()` for hybrid
9//! Rust collection (180K+ SPS on CartPole, zero Python overhead).
10//! - [`dqn`], [`stochastic`], [`deterministic`]: SAC/TD3/DQN network implementations.
11//! - [`mlp`]: Configurable MLP builder with Tanh/ReLU activation.
12
13pub mod actor_critic;
14pub mod collector;
15pub mod continuous_q;
16pub mod convert;
17pub mod deterministic;
18pub mod dqn;
19pub mod entropy;
20pub mod mlp;
21pub mod stochastic;
22mod training;