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
..
mlx	faster group norm (#1304)	2024-08-01 12:49:23 -07:00
src	fix creating array from bf16 tensors in jax / torch (#1305)	2024-08-01 16:20:51 -07:00
tests	Add vmap to scatter (#1200)	2024-08-05 20:12:27 -07:00