Awni Hannun 343e33b6d5 fix all_gather vjp (#2654)

2025-10-07 06:05:23 -07:00

..

mlx

fix cross entropy axis param (#2641)

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

scripts

nccl dep + default for cuda (#2526)

2025-08-21 17:57:49 -07:00

src

Compile now can attach arbitrary data to an entry (#2634)

2025-09-30 13:33:27 -07:00

tests

fix all_gather vjp (#2654)

2025-10-07 06:05:23 -07:00