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