zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
q-sdpa
mlx/benchmarks/python/layer_norm_bench.py
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

1.2 KiB
Raw Permalink Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink