LeonEricsson 6b4b30e3fc Common neural network initializers nn.initializers (#456) ...

* initial commit: constant, normal, uniform

* identity, glorot and he initializers

* docstrings

* rm file

* nits

* nits

* nits

* testing suite

* docs

* nits in docs

* more docs

* remove unused template

* rename packakge to nn.innit

* docs, receptive field

* more docs

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-01-23 06:47:20 -08:00

..

mlx

Common neural network initializers nn.initializers (#456)

2024-01-23 06:47:20 -08:00

src

use axes in tensordot (#525)

2024-01-22 21:17:00 -08:00

tests

Common neural network initializers nn.initializers (#456)

2024-01-23 06:47:20 -08:00