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

_reprlib_fix.py

copyright + ack

2023-11-30 11:12:53 -08:00

extension.py

Add isort pre-commit and run (#68)

2023-12-08 11:31:47 -08:00

optimizers.py

Adds C++ and nn quantization utilities (#230)

2023-12-20 14:17:38 -08:00

utils.py

Minor refactor for tree_map and tree_unflatten (#311)

2023-12-28 20:55:10 -08:00