16 Commits

Author	SHA1	Message	Date
Awni Hannun	29a620cab2	No reshapes in quantized embedding (#1682) ... * no reshapes in quantized embedding * fix inadvertant cast * add tol	2024-12-09 18:57:38 -08:00
Angelos Katharopoulos	5e89aace9b	Fix concatenate vmap (#1600)	2024-11-19 10:44:04 -08:00
Awni Hannun	54f05e7195	Fix gather vmap (#1563) ... * fix gather * fix	2024-11-05 11:29:20 -08:00
Alex Barron	048fabdabd	Fix vmap constant output size (#1524) ... * use inputs to determine output size * remove noop vmap tests	2024-10-30 16:16:53 -07:00
nicolov	8c9f0278b9	Add vmap to scatter (#1200) ... * Add vmap to scatter * updates * vmap updates + a few more tests * bug fix --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-08-05 20:12:27 -07:00
Aneesh Shetty	d0dbfe0b97	Adds radians and degrees (#1011)	2024-04-22 11:17:49 -07:00
Angelos Katharopoulos	3fc993f82d	Properly handle negative axes in python vmap (#944)	2024-04-02 18:07:23 -07:00
nicolov	105d236889	Add vmap for SVD and inverse (#849)	2024-03-21 13:18:27 -07:00
Awni Hannun	19ec023256	vmap matmul and admm (#836)	2024-03-14 14:38:22 -07:00
Angelos Katharopoulos	76c919b4ec	NumberOfElements for shapeless compile and vmap fixes (#802)	2024-03-13 10:34:14 -07:00
Awni Hannun	e88e474fd1	Reduce vmap + some fixes (#601)	2024-02-01 11:30:28 -08:00
Nripesh Niketan	73321b8097	feat: add logicalAnd and logicalOR (#386) ... * feat: add logicalAnd and logicalOR * run pre-commit * Refactor logical_and and logical_or functions * Add acknowledgement * Add logical AND and logical OR operators * Refactor logical_and and logical_or functions * Add support for logical operators on bool arrays * Update mlx/ops.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update mlx/ops.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Add logical AND and OR operators for arrays and scalars * Refactor vjp and jvp methods in primitives.cpp * Add overloaded operators for logical AND and OR * format --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com> Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-08 07:00:05 -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
Zach Schillaci	5b9be57ac3	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -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