7 Commits

Author	SHA1	Message	Date
nicolov	8c9f0278b9	Add vmap to scatter (#1200) ... * Add vmap to scatter * updates * vmap updates + a few more tests * bug fix --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-08-05 20:12:27 -07:00
nicolov	105d236889	Add vmap for SVD and inverse (#849)	2024-03-21 13:18:27 -07:00
Rifur13	126c9869c8	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -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
Awni Hannun	e5851e52b1	Add move and swap axis, and vmap for slice, concat, and gather (#158) ... * add move and swap axis, and vmap for slice, concat, and gather	2023-12-14 12:59:12 -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