Angelos Katharopoulos 110d9b149d Layer norm grad fix donation bug (#941) ...

* add layer norm grad test

* Fix donation bug in layernorm vjp

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-04-01 06:15:50 -07:00

..

mlx

Support Chaining for some of functionalities of nn.Module (#885) (#897)

2024-03-27 19:58:29 -07:00

src

Comparing python objects (such as list/tuple) with mlx.core.array (#920)

2024-03-29 06:52:30 -07:00

tests

Layer norm grad fix donation bug (#941)

2024-04-01 06:15:50 -07:00