113 Commits

Author	SHA1	Message	Date
Awni Hannun	90d04072b7	fix build w/ flatten (#195)	2023-12-17 11:58:45 -08:00
__mo_san__	52e1589a52	implemented Flatten Module (#149) ... * implemented flatten op --------- Co-authored-by: Awni Hannun <awni@apple.com>	2023-12-16 21:54:37 -08:00
Diogo	dc2edc762c	added tri / tril / triu (#170) ... * added tri / tril / triu * fixed tests * ctest tests * tri overload and simplified tests * changes from comment * more tests for m * ensure assert if not 2-D * remove broadcast_to * minor tweaks --------- Co-authored-by: Awni Hannun <awni@apple.com>	2023-12-15 17:30:34 -08:00
Jason	e28b57e371	Added mx.stack c++ frontend impl (#123) ... * stack C++ operation + python bindings	2023-12-14 13:21:19 -08:00
Awni Hannun	e5851e52b1	Add move and swap axis, and vmap for slice, concat, and gather (#158) ... * add move and swap axis, and vmap for slice, concat, and gather	2023-12-14 12:59:12 -08:00
Luca Arnaboldi	b93c4cf378	Floor and Ceil (#150) ... * Implements Floor and Ceil Ops	2023-12-14 10:00:23 -08:00
Cyril Zakka, MD	e080290ba4	Added eye/identity ops (#119) ... `eye` and `identity` C++ and Python ops	2023-12-11 12:38:17 -08:00
Awni Hannun	4e3bdb560c	random generation fix (#80) ... Random generation fix	2023-12-08 10:40:57 -08:00
Awni Hannun	db487e6b1a	format	2023-11-30 11:50:50 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Jagrit Digani	e6306cfee9	jagrit's commit files	2023-11-29 10:52:08 -08:00
Angelos Katharopoulos	d1f86272a2	angelos's commit files	2023-11-29 10:42:59 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00