258 Commits

Author	SHA1	Message	Date
Awni Hannun	e595324804	docs update	2025-08-07 07:46:33 +00:00
Awni Hannun	185a879b05	docs update	2025-08-07 07:46:32 +00:00
Awni Hannun	3df9dab38c	docs update	2025-08-07 07:46:32 +00:00
Awni Hannun	b86da5711b	docs	2025-08-07 07:46:32 +00:00
Awni Hannun	2988d6c14d	docs update	2025-08-07 07:46:32 +00:00
Awni Hannun	680a57bb33	docs update	2025-08-07 07:46:32 +00:00
Awni Hannun	b70128ce73	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	1364ef2259	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	2c0a1a3952	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	aea164fab8	use proper version	2025-08-07 07:46:31 +00:00
Awni Hannun	9abc99cfe9	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	4490e181db	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	e04f658611	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	b04ef9ec37	docs update	2025-08-07 07:46:31 +00:00
Awni Hannun	f2365d5b12	remove uneeded files in docs	2025-08-07 07:46:30 +00:00
Awni Hannun	eea1a5f6ec	update docs	2025-08-07 07:46:30 +00:00
Awni Hannun	1c7ebac536	docs update	2025-08-07 07:46:30 +00:00
Awni Hannun	c07de29458	docs up	2025-08-07 07:46:30 +00:00
Awni Hannun	c0dc02b8f4	docs up	2025-08-07 07:46:30 +00:00
Awni Hannun	e9e7ff09dd	docs update	2025-08-07 07:46:30 +00:00
Awni Hannun	b024ed7744	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	beda33e1a9	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	83d006dfdd	update docs	2025-08-07 07:46:30 +00:00
Awni Hannun	b906475af3	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	0bb6edbe54	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	b728f1fd66	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	3bf234ff88	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	082e939fc7	docs	2025-08-07 07:46:30 +00:00
Awni Hannun	75b55ffa19	docs	2025-08-07 07:46:29 +00:00
Awni Hannun	b8cf6bd778	docs	2025-08-07 07:46:29 +00:00
Awni Hannun	3e96be74e6	docs	2025-08-07 07:46:29 +00:00
Awni Hannun	312037d0f7	docs	2025-08-07 07:46:29 +00:00
Luca Vivona	728d4db582	Support destination arg in tree flatten/unflatten (#2450)	2025-08-06 15:34:59 -07:00
Awni Hannun	db5c7efcf6	revert default cuda install (#2465) ... * revert default cuda install * revert default cuda install	2025-08-06 06:19:12 -07:00
Awni Hannun	6ad0889c8a	default install cuda on linux (#2462)	2025-08-04 15:33:05 -07:00
Awni Hannun	da5912e4f2	fix custom metal extension (#2446)	2025-07-31 06:25:36 -07:00
Awni Hannun	dcb8319f3d	update install docs and requirements (#2419)	2025-07-25 12:13:19 -07:00
Gökdeniz Gülmez	deee214a95	Adding support for the Muon Optimizer (#1914) ... * initial commit with workong optmimizer * update ACKNOWLEDGMENTS.md * nits and adding it to test * nits * G.astype(mx.bfloat16) to G.astype(G.dtype) * G.ndim >= 2 to assert G.ndim == 2 * remove coments * replace with mx.addmm * remove comments * format * nits * match muon * fix addmm --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-07-18 12:25:28 -07:00
Awni Hannun	f0a0b077a0	Install linux with mlx[cuda] and mlx[cpu] (#2356) ... * install linux with mlx[cuda] and mlx[cpu] * temp for testing * cleanup circle, fix cuda repair * update circle * update circle * decouple python bindings from core libraries	2025-07-14 17:17:33 -07:00
Cheng	d34f887abc	Add Primitive::name and remove Primitive::print (#2365)	2025-07-14 14:06:35 -07:00
Cheng	19facd4b20	Build with all cpu cores by default (#2336)	2025-07-07 06:06:45 -07:00
Awni Hannun	76831ed83d	Build CUDA release in Circle (#2306) ... * cuda release * add license	2025-06-19 15:26:36 -07:00
Awni Hannun	c552ff2451	[CUDA] Fix back-end bugs and enable corresponding tests (#2296) ... * Fix some cuda back-end bugs and enable corresponding tests * more fixes * enable more tests * format	2025-06-16 08:45:40 -07:00
Awni Hannun	1ca616844b	Fix unintuitive metal kernel caching (#2242) ... * Fix unintuitive metal kernel caching * alternative solution	2025-06-06 20:08:15 -07:00
Awni Hannun	55b4062dd8	copyright in docs (#2214)	2025-05-21 17:13:04 -07:00
Awni Hannun	a2cadb8218	real and imag properties (#2189)	2025-05-15 18:17:50 -07:00
Awni Hannun	c1eb9d05d9	non-symmetric eig and eigh (#2188)	2025-05-15 13:01:44 -07:00
Aashiq Dheeraj	bb6565ef14	add fftshift and ifftshift fft helpers (#2135) ... * add fftshift and ifftshift fft helpers * address comments * axes have to be iterable * fix fp error in roll + add test --------- Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local>	2025-04-29 22:13:45 -07:00
Yury Popov	e9e268336b	LogCumSumExp (#2069)	2025-04-13 01:27:29 -07:00
Awni Hannun	9c6953bda7	Fix stubgen (#2065) ... * Fix stubgen * add multi optim to docs	2025-04-11 12:02:54 -07:00