mlx_rs/module/
mod.rs

1
2
3
4
5
6
7
8
9
10
11
12
//! This mod defines the traits for neural network modules and parameters.
//!
//! This is to separate the trait definitions from the implementations, which are in the `mlx-nn`
//! crate. This also allows using the `mlx_macros::ModuleParameters` derive macro in crates other
//! than `mlx-nn`.

#[allow(clippy::module_inception)]
mod module;
mod param;

pub use module::*;
pub use param::*;