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
..
nn	Custom primitive + RoPE fat op (#676)	2024-02-14 14:04:25 -08:00
_reprlib_fix.py	copyright + ack	2023-11-30 11:12:53 -08:00
extension.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
optimizers.py	Compile with capture (#629)	2024-02-07 17:29:22 -08:00
py.typed	Add py.typed to support PEP-561 (type-hinting) for mlx (#588)	2024-01-31 12:05:42 -08:00
utils.py	Minor refactor for tree_map and tree_unflatten (#311)	2023-12-28 20:55:10 -08:00