Awni Hannun e88f2d4a8e fix cross entropy axis param (#2641) ...

* fix cross entropy axis param

* faster grad clipping

2025-10-01 16:49:55 -07:00

..

layers

Adding Relu2 (#2582)

2025-09-10 07:24:30 -07:00

__init__.py

Ring docs (#1829)

2025-02-28 11:34:21 -08:00

init.py

catch stream errors earlier to avoid aborts (#1801)

2025-01-27 14:05:43 -08:00

losses.py

fix cross entropy axis param (#2641)

2025-10-01 16:49:55 -07:00

utils.py

Remove stream from average grads so it uses default (#2532)

2025-08-25 15:56:29 -07:00