Tristan Bilot 6022d4129e scatter_max vjp + bindings + tests (#431) ...

Co-authored-by: DjamelMesbah <djamel.mesbah@adservio.fr>

2024-01-14 14:12:15 -08:00

..

mlx

Add type hint for Module (#412)

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

src

Fix stub generation, change graph exporting for arrows to go to outputs (#455)

2024-01-14 14:06:16 -08:00

tests

scatter_max vjp + bindings + tests (#431)

2024-01-14 14:12:15 -08:00