13 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	5c1fa64fb0	Custom transforms (#1246)	2024-07-10 18:00:01 -07:00
Awni Hannun	fd1c08137b	stable cumprod grad at 0 (#1167)	2024-05-31 12:28:42 -07:00
Awni Hannun	e6418781ab	Fix logsumexp edge case (#740) ... * fix logsumexp * fix inf constant * also fix power grad * fix ternary dispatch	2024-02-25 08:39:55 -08:00
Awni Hannun	cb6156d35d	Fix eval in trace bugs (#612) ... * Fix eval in trace bugs * comment nit	2024-02-02 09:57:12 -08:00
Awni Hannun	b207c2c86b	Power VJP fix for 0 (#505)	2024-01-20 01:17:40 -08:00
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