Awni Hannun ccf1645995 Custom primitive + RoPE fat op (#676) ...

* extensions start

* rope custom op

* fix build

* docs + rope benchmark

* fix test

* Add a Metal kernel for RoPE

* Fix position of traditional

* transform tests

* Move rope computation to float and fix tests

* Fix the test and a typo

* change to fast

* fix no metal build

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-02-14 14:04:25 -08:00

..

cpp

Multi output primitives (#330)

2024-01-08 16:39:08 -08:00

numpy

Add isort pre-commit and run (#68)

2023-12-08 11:31:47 -08:00

python

Custom primitive + RoPE fat op (#676)

2024-02-14 14:04:25 -08:00