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 |
|