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

..

__init__.py

Support LR schedulers (#334)

2024-02-15 11:26:20 -08:00

optimizers.py

fix cross entropy axis param (#2641)

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

schedulers.py

refactor: all use schedule (#1973)

2025-03-19 11:24:04 -07:00