1 2 3 4 5 6 7
pub mod dag; /// Contains `Device` and `DeviceBuffer` APIs. pub mod device; /// Contains the raw graph construction and execution API. pub mod graph; /// Contains the `Trainer` struct and `TrainerState` trait. pub mod trainer;