Awni Hannun
|
516ded618b
|
Dynamic slicing (#1741)
* dynamic slice and slice update
* python bindings + tests + fix set item
* fix compile issue
* comment
* fix jit
|
2025-01-07 14:02:16 -08:00 |
|
Awni Hannun
|
e03f0372b1
|
More shape type (#1705)
* more shape type
* fix
|
2024-12-19 08:08:20 -08:00 |
|
Awni Hannun
|
40c62c1321
|
Use int64 stride everywhere (#1671)
* use int64 stride everywhere
* fix ext
* fix ext
* more shape + cleanup
* one more
* few more
|
2024-12-09 11:09:02 -08:00 |
|
fgranqvist
|
50eff6a10a
|
Implement sampling from laplace distribution. (#1279)
|
2024-07-24 15:15:37 +02:00 |
|
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 |
|
Awni Hannun
|
bc06cb9ff6
|
Pickle + dtype fix for numpy conversion (#763)
* pickle + dtype fix for numpy conversion
* fix getattribute on Module base
* remove unused function
* fix tests
* add topk to ops
* fix doc
|
2024-03-02 06:09:29 -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 |
|
Awni Hannun
|
4e3bdb560c
|
random generation fix (#80)
Random generation fix
|
2023-12-08 10:40:57 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Awni Hannun
|
8ca7f9e8e9
|
awni's commit files
|
2023-11-29 10:30:41 -08:00 |
|