mlx_rs/nn/
mod.rs

1#![deny(missing_docs, missing_debug_implementations)]
2
3//! Neural network support for MLX
4//!
5//! All modules provide a `new()` function that take mandatory parameters and other methods
6//! to set optional parameters.
7
8mod activation;
9mod container;
10mod convolution;
11mod convolution_transpose;
12mod dropout;
13mod embedding;
14mod linear;
15mod normalization;
16mod pooling;
17mod positional_encoding;
18mod quantized;
19mod recurrent;
20mod transformer;
21mod upsample;
22mod value_and_grad;
23
24pub use activation::*;
25pub use container::*;
26pub use convolution::*;
27pub use convolution_transpose::*;
28pub use dropout::*;
29pub use embedding::*;
30pub use linear::*;
31pub use normalization::*;
32pub use pooling::*;
33pub use positional_encoding::*;
34pub use quantized::*;
35pub use recurrent::*;
36pub use transformer::*;
37pub use upsample::*;
38pub use value_and_grad::*;
mlx_rs/nn/mod.rs

mlx_rs/nn/
mod.rs
