Awni Hannun a2bf7693dd Primitive's VJP takes outputs as input (#475) ...

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-01-16 19:03:53 -08:00

..

mlx

Add type hint for Module (#412)

2024-01-10 11:23:42 -08:00

src

implemented isposinf and isneginf in one PR (#470)

2024-01-16 06:48:07 -08:00

tests

Primitive's VJP takes outputs as input (#475)

2024-01-16 19:03:53 -08:00