Angelos Katharopoulos 5adf185f86 Fix update_modules() when providing a subset (#2308)

2025-06-20 17:19:46 -07:00

..

layers

Fix update_modules() when providing a subset (#2308)

2025-06-20 17:19:46 -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

update the formula of smooth_l1_loss (#1986)

2025-03-21 06:25:23 -07:00

utils.py

Data parallel helper (#1407)

2024-09-16 18:17:21 -07:00