10 Commits

Author	SHA1	Message	Date
Jagrit Digani	639e06e1f3	Indexing bug fix (#947) ... * Fix axes accounting * Add tests	2024-04-01 12:18:50 -07:00
Abdussamet Türker	5611e1a95e	Fix unsqueeze with None (#899) ... * Fix unsqueeze with None * Clean unnecessary files	2024-03-26 13:59:44 -07:00
Jagrit Digani	8e5a5a1ccd	Set item bug fix (#879) ... * set item shaping bug fix * Add extra tests	2024-03-22 12:11:17 -07:00
Jagrit Digani	a5681ebc52	Update set item (#861) ... * Update mlx_set_item to handle regular slices without expanding * Refactor ellipsis handling * Route mlx_set_item to slice_update where possible * Update mlx_scatter_args_slice * Don't route to gather if no array indices	2024-03-21 02:48:13 -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
Angelos Katharopoulos	961435a243	Scatter vjp (#394) ... * Add a first scatter vjp * Implement the scatter_add vjp * Add array.at to implement user friendly scatters	2024-01-09 13:36:51 -08:00
Awni Hannun	f40d17047d	Indexing bug (#233) ... * fix * test	2023-12-20 10:44:01 -08:00
Awni Hannun	104c34f906	setite negative indexing bug (#189)	2023-12-16 06:44:47 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00