mlx/python
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