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