mlx/docs/src/python
Justin Deschenaux 8e5600022a
Implement RNN, GRU, LSTM (#268)
* RNN base implementation

* Address comments+format

* nits in docs

* add tests for prb

* fix test

* add a couple tests

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-03-11 21:14:44 -07:00
..
nn Implement RNN, GRU, LSTM (#268) 2024-03-11 21:14:44 -07:00
optimizers Add linear warmup and schedule joining for use with existing schedules (#721) 2024-02-26 07:28:48 -08:00
array.rst Adds round op and primitive (#203) 2023-12-18 11:32:48 -08:00
data_types.rst Fix data_types.rst uint64 (#406) 2024-01-09 06:40:10 -08:00
devices_and_streams.rst Adds device context manager (#679) 2024-02-14 14:14:58 -08:00
fft.rst awni's commit files 2023-11-29 10:30:41 -08:00
linalg.rst QR factorization (#310) 2024-01-26 09:27:31 -08:00
metal.rst bindings for memory info (#761) 2024-03-01 19:51:58 -08:00
nn.rst Common neural network initializers nn.initializers (#456) 2024-01-23 06:47:20 -08:00
ops.rst Pickle + dtype fix for numpy conversion (#763) 2024-03-02 06:09:29 -08:00
optimizers.rst Support LR schedulers (#334) 2024-02-15 11:26:20 -08:00
random.rst Sort some APIs docs by names (a-z) (#472) 2024-01-16 19:37:50 -08:00
transforms.rst Compile docs (#653) 2024-02-08 11:21:50 -08:00
tree_utils.rst jagrit's commit files 2023-11-29 10:52:08 -08:00