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
..
cpp	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
numpy	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
python	Implement vjps for some primitives in the fast namespace (#883)	2024-03-26 16:35:34 -07:00