8 Commits

Author	SHA1	Message	Date
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