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
..
layers	feat: Add Dropout3d layer to nn.layers (#313)	2023-12-31 14:01:21 -08:00
__init__.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
losses.py	implement-batch-norm-layer (#217)	2023-12-25 07:32:53 -08:00
utils.py	copyright + ack	2023-11-30 11:12:53 -08:00