rlox_nn/
lib.rs

1//! rlox-nn: Backend-agnostic neural network traits for reinforcement learning.
2//!
3//! Defines the [`ActorCritic`], [`QFunction`], [`StochasticPolicy`],
4//! [`DeterministicPolicy`], and [`EntropyTuner`] traits that any NN backend
5//! (Candle, Burn, or custom) can implement.
6//!
7//! Key types:
8//! - [`TensorData`]: Flat `Vec<f32>` with shape metadata, safe for FFI.
9//! - [`ActionOutput`], [`EvalOutput`]: Structured policy outputs.
10//! - [`PPOStepConfig`], [`DQNStepConfig`]: Algorithm hyperparameters.
11
12pub mod distributions;
13pub mod error;
14pub mod tensor_data;
15pub mod traits;
16
17pub use error::NNError;
18pub use tensor_data::TensorData;
19pub use traits::*;