5 Commits

Author	SHA1	Message	Date
Awni Hannun	33421c1dd3	Limit grad recursion depth by not recursing through non-grad inputs (#1764) ... * limit grad recursion depth * add grad of module test	2025-01-14 14:33:18 -08:00
Cheng	0bf19037ca	Remove "using namespace mlx::core" in python/src (#1689)	2024-12-11 15:45:39 -08:00
Angelos Katharopoulos	3fc993f82d	Properly handle negative axes in python vmap (#944)	2024-04-02 18:07:23 -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
Awni Hannun	fe1dabf272	Fix compile with non standard types (#745) ... * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-02-26 19:28:53 -08:00