Awni Hannun 29a620cab2 No reshapes in quantized embedding (#1682) ...

* no reshapes in quantized embedding

* fix inadvertant cast

* add tol

2024-12-09 18:57:38 -08:00

..

layers

No reshapes in quantized embedding (#1682)

2024-12-09 18:57:38 -08:00

__init__.py

Common neural network initializers nn.initializers (#456)

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

init.py

[Feature] Added Sparse Initialization (#1498)

2024-10-24 12:31:24 -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