zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-23 05:58:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
29221fa23895f6d3770b73ed8bb5105b6f9a7fe1
mlx/benchmarks
History
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