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