Angelos Katharopoulos 29221fa238 Implement vjps for some primitives in the fast namespace (#883) ...

* Implement rope vjp in terms of rope
* RMSNormVJP primitive and kernel
* Add LayerNormVJP primitive and kernel

2024-03-26 16:35:34 -07:00

..

mlx

pick up preivously set attributes (#905)

2024-03-26 11:19:59 -07:00

src

Fix unsqueeze with None (#899)

2024-03-26 13:59:44 -07:00

tests

Implement vjps for some primitives in the fast namespace (#883)

2024-03-26 16:35:34 -07:00