rogue_net/
lib.rs

1mod categorical_action_head;
2mod config;
3mod embedding;
4mod fun;
5mod layer_norm;
6mod linear;
7mod msgpack;
8mod relpos_encoding;
9mod rogue_net;
10mod state;
11#[cfg(test)]
12mod tests;
13mod transformer;
14
15pub use crate::config::RogueNetConfig;
16pub use crate::rogue_net::{FwdArgs, RogueNet};