mlx_rs/module/
mod.rs

1//! This mod defines the traits for neural network modules and parameters.
2//!
3//! This is to separate the trait definitions from the implementations, which are in the `mlx-nn`
4//! crate. This also allows using the `mlx_macros::ModuleParameters` derive macro in crates other
5//! than `mlx-nn`.
6
7#[allow(clippy::module_inception)]
8mod module;
9mod param;
10
11pub use module::*;
12pub use param::*;