Awni Hannun
|
2419edd5b2
|
Faster indexing math in a few kernels (#1589)
* wip: faster compiled kernels
* faster general unary with uint specialization
* index type in compiled, unary, binary, ternary, copy
* fix jit
* jit fix
* specialize gather + scatter
* nit in docs
|
2024-11-18 19:52:00 -08:00 |
|
Chris Offner
|
46d8b16ab4
|
Fix vmap example in docs (#1556)
|
2024-11-02 17:44:14 -07:00 |
|
Chris Offner
|
42533931fa
|
Fix typo "it's" -> "its" (#1555)
|
2024-11-02 06:06:34 -07:00 |
|
Awni Hannun
|
c26208f67d
|
Remove Hazard tracking with Fences (#1509)
* remove hazard tracking
* with fence map
* no hazard tracking with fences
* nits
* fix fence retain
* cleanup
* fix quantized rebase
|
2024-10-21 19:33:32 -07:00 |
|
Angelos Katharopoulos
|
0163a8e57a
|
Add docs for the distributed namespace (#1184)
|
2024-06-06 11:37:00 -07:00 |
|
K Venkat Ramnan
|
ab977109db
|
feat: Added dlpack device (#1165)
* feat: Added dlpack device
* feat: Added device_id to dlpack device
* feat: Added device_id to dlpack device
* doc: updated conversion docs
* doc: updated numpy.rst dlpack information
* doc: updated numpy.rst dlpack information
* Update docs/src/usage/numpy.rst
* Update docs/src/usage/numpy.rst
---------
Co-authored-by: Venkat Ramnan Kalyanakumar <venkatramnankalyanakumar@Venkats-MacBook-Air.local>
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
|
2024-05-31 12:29:01 -07:00 |
|
Alex Shepard
|
91eba8e485
|
fix for grammatical typo in docs (#988)
thanks for mlx!
|
2024-04-11 17:02:06 -07:00 |
|
Suvan Kumar
|
433c0206b0
|
Update saving_and_loading.rst (#929)
Update saving / load docs.
|
2024-03-30 14:30:06 -07:00 |
|
AlexCheema
|
7762e07fde
|
Update function_transforms.rst (#796)
Fix typo in function_transforms.rst
|
2024-03-06 12:03:37 -08:00 |
|
Awni Hannun
|
5c03efaf29
|
Compile docs (#653)
* compile docs
* docs nits + comments
|
2024-02-08 11:21:50 -08:00 |
|
Awni Hannun
|
09b9275027
|
Make shape a tuple (#591)
* shape tuple
* also remove simplify from docs
* rebase
|
2024-01-30 13:11:01 -08:00 |
|
Awni Hannun
|
78e5f2d17d
|
usage doc for function transformations (#481)
|
2024-01-17 17:10:53 -08:00 |
|
Awni Hannun
|
c92a134b0d
|
more docs (#421)
* more docs
* fix link
* nits + comments
|
2024-01-10 14:04:12 -08:00 |
|
Daniel Strobusch
|
1331fa19f6
|
Make array conform to the Python Buffer Protocol (#323)
|
2024-01-05 15:58:33 -08:00 |
|