mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-29 22:01:17 +08:00
fffe072028
* Implementation of mlx.random.multivariate_normal (#502) * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Updated typo in docstring * Restricted multivariate_normal to float32 * Generic mean and variance shapes * Review edits * Update mlx/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Update python/src/random.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * Test for ndim of mean and cov * nits * smaller size for test * fix broadcasted sampling --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com> Co-authored-by: Awni Hannun <awni@apple.com> |
||
|---|---|---|
| .. | ||
| 3rdparty | ||
| backend | ||
| io | ||
| types | ||
| allocator.cpp | ||
| allocator.h | ||
| array.cpp | ||
| array.h | ||
| CMakeLists.txt | ||
| compile_impl.h | ||
| compile.cpp | ||
| compile.h | ||
| device.cpp | ||
| device.h | ||
| dtype.cpp | ||
| dtype.h | ||
| fast_primitives.h | ||
| fast.cpp | ||
| fast.h | ||
| fft.cpp | ||
| fft.h | ||
| graph_utils.cpp | ||
| graph_utils.h | ||
| io.h | ||
| linalg.cpp | ||
| linalg.h | ||
| mlx.h | ||
| ops.cpp | ||
| ops.h | ||
| primitives.cpp | ||
| primitives.h | ||
| random.cpp | ||
| random.h | ||
| scheduler.cpp | ||
| scheduler.h | ||
| stream.h | ||
| transforms_impl.h | ||
| transforms.cpp | ||
| transforms.h | ||
| utils.cpp | ||
| utils.h | ||