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