Luca Arnaboldi fffe072028 Implementation of mlx.random.multivariate_normal (#502) (#877) ... * 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>		2024-04-09 13:50:12 -07:00
..
array.cpp	Better exceptions in case of invalid operations on mlx.core.array (#910) (#926)	2024-04-02 21:11:24 -07:00
buffer.h	Remove unnecessary string copies (#891)	2024-03-28 13:14:59 -07:00
CMakeLists.txt	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
constants.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
convert.cpp	Ensure shape dimensions are within supported integer range (#566) (#704)	2024-03-25 13:29:45 -07:00
convert.h	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
device.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
fast.cpp	post nanobind docs fixes and some updates (#889)	2024-03-24 15:03:27 -07:00
fft.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
indexing.cpp	Indexing bug fix (#947)	2024-04-01 12:18:50 -07:00
indexing.h	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
linalg.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
load.cpp	Do not store iostream in shared_ptr (#872)	2024-03-22 06:54:45 -07:00
load.h	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
metal.cpp	Improve profiling with gpu tracing (#969)	2024-04-07 21:47:43 -07:00
mlx.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
ops.cpp	Add mx.meshgrid (#961)	2024-04-09 11:43:08 -07:00
random.cpp	Implementation of mlx.random.multivariate_normal (#502) (#877)	2024-04-09 13:50:12 -07:00
stream.cpp	Switch to nanobind (#839)	2024-03-18 20:12:25 -07:00
transforms.cpp	Properly handle negative axes in python vmap (#944)	2024-04-02 18:07:23 -07:00
trees.cpp	Properly handle negative axes in python vmap (#944)	2024-04-02 18:07:23 -07:00
trees.h	Properly handle negative axes in python vmap (#944)	2024-04-02 18:07:23 -07:00
utils.h	Better exceptions in case of invalid operations on mlx.core.array (#910) (#926)	2024-04-02 21:11:24 -07:00