mlx/docs/src/python
Nripesh Niketan e09bf35b28
feat: Add Dropout3d layer to nn.layers (#313)
* feat: Add Dropout3d layer to nn.layers

* acknowledgement

* Add dropout tests to test_nn.py

* run pre-commit

* Add activation functions and dropout3d ops

* Add dropout tests for bfloat16 and float16
2023-12-31 14:01:21 -08:00
..
nn feat: Add Dropout3d layer to nn.layers (#313) 2023-12-31 14:01:21 -08:00
array.rst Adds round op and primitive (#203) 2023-12-18 11:32:48 -08:00
data_types.rst jagrit's commit files 2023-11-29 10:52:08 -08:00
devices_and_streams.rst jagrit's commit files 2023-11-29 10:52:08 -08:00
fft.rst awni's commit files 2023-11-29 10:30:41 -08:00
linalg.rst linalg.norm (#187) 2023-12-26 19:42:04 -08:00
nn.rst Add Hinge, Huber and LogCosh losses (#199) 2023-12-22 10:28:10 -08:00
ops.rst Add support for repeat (#278) 2023-12-27 13:11:38 -08:00
optimizers.rst Add Lion optimizer (#209) 2023-12-20 13:54:58 -08:00
random.rst awni's commit files 2023-11-29 10:30:41 -08:00
transforms.rst Update docs (#177) (#190) 2023-12-16 06:52:18 -08:00
tree_utils.rst jagrit's commit files 2023-11-29 10:52:08 -08:00