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