Lucas Newman 4a64d4bff1 Add support for grouped 1D convolutions to the nn API (#1444) ... * Fix the weight shape for grouped convolutions from the nn API. * Add tests. * Pre-commit formatting. * Add input validation. * Use integer division instead of casting. * docs * nit --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-09-28 06:41:07 -07:00
..
layers	Add support for grouped 1D convolutions to the nn API (#1444)	2024-09-28 06:41:07 -07:00
__init__.py	Common neural network initializers nn.initializers (#456)	2024-01-23 06:47:20 -08:00
init.py	Some fixes to typing (#1371)	2024-08-28 11:16:19 -07:00
losses.py	Some fixes to typing (#1371)	2024-08-28 11:16:19 -07:00
utils.py	Data parallel helper (#1407)	2024-09-16 18:17:21 -07:00