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