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