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