1481 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Ronan Collobert	8d13a0bc6b	WIP (metal)	2025-10-31 16:24:21 -07:00
Ronan Collobert	ac75c87fd7	WIP (cpu)	2025-10-31 16:24:09 -07:00
Ronan Collobert	7107802e09	WIP (examples)	2025-10-31 16:23:51 -07:00
Ronan Collobert	c5913131cf	WIP (distributed)	2025-10-31 13:32:56 -07:00
Ronan Collobert	19ab7911f6	WIP (cuda)	2025-10-31 13:32:43 -07:00
Ronan Collobert	4a1b1796b7	WIP (io)	2025-10-31 13:20:47 -07:00
Ronan Collobert	b48d298205	WIP (distributed)	2025-10-31 13:20:09 -07:00
Ronan Collobert	8277e71ea9	WIP (gpu)	2025-10-31 13:19:54 -07:00
Ronan Collobert	b0d985416a	fix arg_reduce	2025-10-31 13:13:15 -07:00
Ronan Collobert	8d10f3ec75	WIP (metal)	2025-10-31 11:47:03 -07:00
Ronan Collobert	6343622c67	fix small vector indexing checks	2025-10-31 11:46:36 -07:00
Ronan Collobert	979abf462b	WIP (metal)	2025-10-31 09:43:29 -07:00
Ronan Collobert	981d2fdaf0	WIP (cpu)	2025-10-31 09:40:50 -07:00
Ronan Collobert	5a306d3495	WIP (common)	2025-10-31 09:40:13 -07:00
Ronan Collobert	5baa361779	WIP (tests)	2025-10-31 09:39:38 -07:00
Ronan Collobert	1bac0db7e3	WIP	2025-10-30 16:25:36 -07:00
Ronan Collobert	a1212b4e44	WIP (distributed)	2025-10-30 16:25:11 -07:00
Ronan Collobert	45a8b226af	WIP (cpu)	2025-10-30 16:24:51 -07:00
Ronan Collobert	76ef1e98f3	WIP (common)	2025-10-30 16:18:59 -07:00
Ronan Collobert	63d91557e0	fix FFT (PocketFFT requires size_t for axis)	2025-10-29 17:05:48 -07:00
Ronan Collobert	310e501e6a	WIP (cpu)	2025-10-29 16:52:25 -07:00
Ronan Collobert	cacc3ab7fd	WIP (common)	2025-10-29 16:51:42 -07:00
Ronan Collobert	53525cba23	WIP	2025-10-29 16:51:05 -07:00
Ronan Collobert	3d67b717a0	the cpu simd case	2025-10-29 16:43:18 -07:00
Ronan Collobert	953b2f5be2	WIP	2025-10-29 16:11:32 -07:00
Ronan Collobert	26f7155537	SmallVector: keep sizes small (int)	2025-10-29 16:06:10 -07:00
Ronan Collobert	66fcb9fe94	array: use int or int64_t instead of size_t	2025-10-29 16:04:04 -07:00
Awni Hannun	d1e06117e8	bump python (#2694)	2025-10-27 11:34:31 -07:00
Awni Hannun	539d8322d1	add median op (#2705)	2025-10-27 11:33:42 -07:00
Awni Hannun	c4767d110f	fix addmm cpu (#2699)	2025-10-27 11:33:32 -07:00
David Koski	895217f25b	optionally load metallib from framework (#2702) ... * optionally load metallib from framework * pre-commit * adjust logic	2025-10-27 07:52:03 -07:00
Manuel Villanueva	0cfeeb60ca	Einsum error msg improvement (#2690) ... * Improved error message for Einsum * Modifications via pre-commit * format * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-27 06:31:47 -07:00
Ronan Collobert	8f8af61a37	fix warnings showing up with -Wall (#2692)	2025-10-24 11:43:35 -07:00
Manuel Villanueva	233384161e	Improved mx.split() docs (#2689) ... * Improved mx.split() documentation * Fix typo in docstring for array split function * add example --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-24 09:48:41 -07:00
Awni Hannun	5bcf3a6794	format	2025-10-22 16:08:47 -07:00
wickedcoder	7707196297	Merge commit from fork ... * add length validation to the header * fix accessing out of bound index with .at()	2025-10-22 15:31:25 -07:00
wickedcoder	7e3471c987	Merge commit from fork ... * add tensor->weights_data validation * add null pointer check for tensor	2025-10-22 15:31:03 -07:00
Awni Hannun	9f0ba3ddf1	patch bump (#2680)	2025-10-17 12:12:07 -07:00
Awni Hannun	4bce5f9b2d	suppress gcc 10.1 warnings (#2679) ... * suppress gcc 10.1 warnings * suppress gcc 10.1 warnings v0.29.3	2025-10-17 12:09:21 -07:00
Anastasiia Filippova	e9eab527eb	Nccl timeout (#2673) ... * print the error & delete nccl group * timeout for nccl binding * typo * revert error * fixed a typo	2025-10-14 12:29:54 -07:00
Awni Hannun	36ca62dba8	remove unused unary file (#2672)	2025-10-13 19:36:26 -07:00
Manuel Villanueva	9cbb1b0148	Modified sort behavior when running CPU or Metal to match NumPy/JAX (#2667) ... * Modified sort behavior when running CPU or Metal to match NumPy/JAX sorting behavior. * Modified sort behavior when running CPU or Metal to match NumPy/JAX * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2025-10-13 14:36:45 -07:00
Fabrizio Milo	9bfc476d72	Normalize README bullet formatting (#2671)	2025-10-13 12:13:30 -07:00
Awni Hannun	25e2356316	speed up scalars (#2669)	2025-10-13 12:10:15 -07:00
Awni Hannun	226a1d24e0	Debug cuda conv (#2662) ... * use t4 * use t4	2025-10-10 16:12:47 -07:00
Awni Hannun	630350ad3e	Precise sigmoid (#2659) ... * bump patch * Sigmoid matches PyTorch and is more precise on tails	2025-10-10 10:05:23 -07:00
Awni Hannun	380aeb58ae	enable admm low-precision cpu (#2661)	2025-10-10 09:50:54 -07:00
Awni Hannun	f37389d100	bump patch (#2658)	2025-10-10 08:36:41 -07:00
Awni Hannun	e89e8b4272	Export with callback (#2612) ... * export with callback * export with callback * Add types, fix kwarg ordering bug + test * cleanup, test, fix * typos	2025-10-08 19:24:33 -07:00
AN Long	85a8824a8c	Fix cumulative operations when axis=None (#2653)	2025-10-08 15:25:38 -07:00