7 Commits

Author	SHA1	Message	Date
Noah Farr	5fd11c347d	Add loc and scale to random.normal (#638) ... * Add loc and scale to random.normal * Add tests for loc and scale for random.normal * Run pre-commit hooks * Fix code review	2024-02-07 11:49:59 -08:00
Awni Hannun	09b9275027	Make shape a tuple (#591) ... * shape tuple * also remove simplify from docs * rebase	2024-01-30 13:11:01 -08:00
Daniel Strobusch	dfdb284e16	make behaviour of dtype arguments consistent and compliant to numpy (#379) ... All functions that take an optional dtype should * have a default dtype visible in the generated docs (accomplished via `"dtype"_a = std::optional{float32}`) * behave identical when `dtype=None` or no dtype is passed This important when passing kw args down from a numpy function like: ``` def f(x, dtype=None): mx.random.uniform(dtype=dtype) # ... ``` NumPy functions behave like this. It also fixes a minor bug in `tri`: #378 Closes #378	2024-01-05 09:37:46 -08:00
Daniel Strobusch	7365d142a3	random.uniform must respect dtype, even if lower precision than "low" (#280) ... Fix an edge case where random uniform returns a float32 array, even if a lower precision dtype is wanted due to adding the float32 "low" array.	2023-12-24 07:04:43 -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