mlx-nn 0.1.0

Neural network modules: Linear, LayerNorm, Attention, etc.
Documentation