mutexuan
|
350095ce6e
|
fix type cast error in item() for bfloat16 (#339)
Co-authored-by: xuan <xuan@apple.com>
|
2024-01-01 19:02:04 -08:00 |
|
Daniel Strobusch
|
d58ac083f3
|
expose itemsize and nbytes as for numpy arrays (#284)
see:
* https://numpy.org/doc/stable/reference/generated/numpy.ndarray.nbytes.html
* https://numpy.org/doc/stable/reference/generated/numpy.ndarray.itemsize.html
relates to https://github.com/ml-explore/mlx-examples/pull/174
|
2023-12-25 10:34:28 -08:00 |
|
Angelos Katharopoulos
|
2807c6aff0
|
Implements divide for integer types and adds floor_divide op (#228)
* Add floor_divide
* Add floor_divide to the tests
* Add floor_divide to the docs
|
2023-12-19 20:12:19 -08:00 |
|
Angelos Katharopoulos
|
4d4af12c6f
|
Adds round op and primitive (#203)
|
2023-12-18 11:32:48 -08:00 |
|
__mo_san__
|
52e1589a52
|
implemented Flatten Module (#149)
* implemented flatten op
---------
Co-authored-by: Awni Hannun <awni@apple.com>
|
2023-12-16 21:54:37 -08:00 |
|
Awni Hannun
|
e5851e52b1
|
Add move and swap axis, and vmap for slice, concat, and gather (#158)
* add move and swap axis, and vmap for slice, concat, and gather
|
2023-12-14 12:59:12 -08:00 |
|
Awni Hannun
|
25f70d4ca4
|
Fix divide types + floor divide (//) (#138)
* divide types
* fix black + test
|
2023-12-11 20:20:58 -08:00 |
|
Angelos Katharopoulos
|
3214629601
|
Mlx array accessor (#128)
* Add an accessor to interoperate with custom types
* Change the docs to custom signatures
|
2023-12-11 13:42:55 -08:00 |
|
Angelos Katharopoulos
|
fd836d891b
|
Hashable dtype and mlx.core prefixed repr (#89)
* Make dtype hashable
* Add mlx.core prefix to our dtypes' repr
* Update the dtype test
|
2023-12-09 09:35:28 -08:00 |
|
Angelos Katharopoulos
|
2b714714e1
|
Add the remainder op (#85)
* Add remainder in the C++ backend
* Add the python binding and test
|
2023-12-08 15:08:52 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Angelos Katharopoulos
|
d1f86272a2
|
angelos's commit files
|
2023-11-29 10:42:59 -08:00 |
|