mlx/docs/src/python/nn
Awni Hannun cd9e184529
Quantize embedding (#994)
* quantize embedding

* rename as_linear + comment

* consistency in docs

* fix test
2024-04-15 16:42:10 -07:00
..
functions.rst Fix compile with non standard types (#745) 2024-02-26 19:28:53 -08:00
init.rst Common neural network initializers nn.initializers (#456) 2024-01-23 06:47:20 -08:00
layers.rst Quantize embedding (#994) 2024-04-15 16:42:10 -07:00
losses.rst Add Margin Ranking Loss (#536) 2024-02-02 10:57:31 -08:00
module.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00