mlx-nn 0.1.0

Neural network modules: Linear, LayerNorm, Attention, etc.
Documentation
  • Coverage
  • 100%
    9 out of 9 items documented0 out of 2 items with examples
  • Size
  • Source code size: 33.31 kB This is the summed size of all the files inside the crates.io package for this release.
  • Documentation size: 2.87 MB This is the summed size of all files generated by rustdoc for all configured targets
  • Ø build duration
  • this release: 31s Average build duration of successful builds.
  • all releases: 31s Average build duration of successful builds in releases after 2024-10-23.
  • Links
  • crates.io
  • Dependencies
  • Versions
  • Owners
  • community-stevedores-org

Neural network modules for MLX.

Provides common building blocks: Linear, Embedding, LayerNorm, and RMSNorm. Each module stores its parameters as Tensor values and exposes a forward() method.