Commit Graph

9 Commits

Author SHA1 Message Date
Cheng fd27829efa Build and test python package on Windows CI (#3049) 2026-01-24 07:22:36 +09:00
Awni Hannun 17310d91a6 Add batch offsets for mx.fast.rope (#2564)
* implement batch rope for Metal

* cuda rope (#2576)
2025-09-08 17:35:07 -07:00
Awni Hannun 5580b47291 iinfo and scalar overflow detection (#2009) 2025-03-27 19:54:56 -07:00
Alex Barron 55c5ac7820 fix int64 bug (#1860) 2025-02-12 19:23:46 -08:00
Awni Hannun f110357aaa Bump nanobind to 2.4 + fix (#1710)
* bump nanobind to 2.4 + fix

* fix
2024-12-17 10:57:54 -08:00
Cheng 0bf19037ca Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
Awni Hannun 9814a2ae12 fix conversion to array (#1070) 2024-05-06 16:02:49 -07:00
Awni Hannun 9a8ee00246 Switch to nanobind (#839)
* mostly builds

* most tests pass

* fix circle build

* add back buffer protocol

* includes

* fix for py38

* limit to cpu device

* include

* fix stubs

* move signatures for docs

* stubgen + docs fix

* doc for compiled function, comments
2024-03-18 20:12:25 -07:00
Diogo 35431a4ac8 Adds device context manager (#679) 2024-02-14 14:14:58 -08:00