a828e769be
GPU discovery ( #3055 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2026-01-26 09:54:13 -08:00
c35f4d089a
start cuda circle config ( #2256 )
...
* rebase
* fix metal kernel linking issue on cuda
* start cuda circle config
2025-06-10 21:19:47 -07:00
0bf19037ca
Remove "using namespace mlx::core" in python/src ( #1689 )
2024-12-11 15:45:39 -08:00
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
35431a4ac8
Adds device context manager ( #679 )
2024-02-14 14:14:58 -08:00
41cc7bdfdb
Fix stub generation, change graph exporting for arrows to go to outputs ( #455 )
2024-01-14 14:06:16 -08:00
46a39e5b1f
copyright + ack
2023-11-30 11:12:53 -08:00
d1f86272a2
angelos's commit files
2023-11-29 10:42:59 -08:00
Daniel Hiltgen