Chaoran Yu b054838780 Added clarification to apply_fn parameter of apply_to_modules (#2831) ...

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

2025-11-26 15:40:56 -08:00

..

layers

Added clarification to apply_fn parameter of apply_to_modules (#2831)

2025-11-26 15:40:56 -08: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