8 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	d8fabaa12b	Split multi output (#461) ... * Multi-output split primitive * Add the multi-output split to the ArrayIterator * Add some grad tests for split	2024-01-16 13:33:55 -08:00
Tristan Bilot	f44c132f4a	Add scatter_min VJP (#462)	2024-01-16 00:37:40 -08:00
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
Awni Hannun	3b4f066dac	Correct types for vjp + tests (#418) ... * correct types for vjp + tests * fix build + comment	2024-01-10 13:32:37 -08:00
Angelos Katharopoulos	a611b0bc82	Removes the retain_graph flag (#385) ... * Adds global tracing flag * Removes retain_graph in favor of is_tracer	2024-01-07 15:16:51 -08:00
Zach Schillaci	5b9be57ac3	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00