1 Commits

Author	SHA1	Message	Date
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