L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
1,738 Commits 15 Branches 83 Tags
ce45c52505c8158ea48d2a54e8caae05efd86bfe
Commit Graph

37 Commits

Author SHA1 Message Date
Alex Skryl f2f2d16451 Export: preserve Dtype state values in export callback arguments (#3145) 
Co-authored-by: Awni Hannun <awni@apple.com>
 2026-02-19 08:07:28 -08:00
stef c86a9bced1 [Docs] Simple example of using MLX distributed (#2973) 
Co-authored-by: Awni Hannun <awni@apple.com>
 2026-01-28 17:14:56 -08:00
Angelos Katharopoulos 3fe7794f22 Reverts changing the MLX_IBV_DEVICES to MLX_JACCL_DEVICES (#2999) 2026-01-14 15:44:17 -08:00
Evan Quiney a8197795f5 replace MLX_IBV_COORDINATOR with MLX_JACCL_COORDINATOR (#2986) 2026-01-13 11:26:25 -08:00
Angelos Katharopoulos 9194ec20a8 Thunderbolt RDMA communications backend (#2808) 2025-12-17 11:27:54 -08:00
Satyam singh 6b330eb2d5 DOC : Add compile state example (#2910) 2025-12-16 06:32:58 -08:00
Satyam singh f7a400951a Fix docs: replace mx.random.randn with mx.random.normal (#2890) 2025-12-09 11:46:30 -08:00
CCYeh 8879ee00eb Support more Numpy interfaces for masked_scatter (#2832) 2025-12-01 17:51:02 -08:00
CCYeh b3825ac149 Add Masked Scatter (#2663) 
Co-authored-by: Awni Hannun <awni@apple.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
 2025-11-19 14:53:32 -08:00
Pedro Cuenca eba6a9d163 Compatibility with pip-installed openmpi (#2741) 2025-11-07 16:58:31 -08:00
Umberto Mignozzetti 3f730e77aa Update export function example for array input (#2598) 
After changing the shape to conform (same shapes for all objects), the example works.
 2025-09-16 14:38:05 -07:00
Umberto Mignozzetti 8afb6d62f2 Fix typo in average_gradients function call (#2594) 2025-09-15 11:29:21 -07:00
Umberto Mignozzetti 36cad99a11 Refactor code examples to use 'gelu' (#2592) 
Updated code examples to use 'gelu' directly instead of 'nn.gelu'.
 2025-09-15 09:47:02 -07:00
Krishi Saripalli c5d2937aa5 chore: Update Docs With Slice Copy Example (#2559) 
* chore: updated docs with slice copy example

* nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>
 2025-09-02 22:07:02 -07:00
russellizadi 512281781c Remove state return from function example in compile documentation (#2518) 2025-08-20 00:45:05 -07:00
Luca Vivona 728d4db582 Support destination arg in tree flatten/unflatten (#2450) 2025-08-06 15:34:59 -07:00
Awni Hannun c552ff2451 [CUDA] Fix back-end bugs and enable corresponding tests (#2296) 
* Fix some cuda back-end bugs and enable corresponding tests

* more fixes

* enable more tests

* format
 2025-06-16 08:45:40 -07:00
Angelos Katharopoulos 5d68082881 Ring docs (#1829) 2025-02-28 11:34:21 -08:00
Awni Hannun 1c0c118f7c Fp64 on the CPU (#1843) 
* add fp64 data type

* clean build

* update docs

* fix bug
 2025-02-07 15:52:22 -08:00
Awni Hannun 7064fed1b1 Minor update on MPI docs (#1805) 2025-01-28 11:00:08 -08:00
Awni Hannun b51d70a83c export docs (#1747) 2025-01-03 15:04:17 -08:00
Awni Hannun ae69cb15e9 shapeless compile in docs and partially shapeless reshape (#1742) 2025-01-02 16:24:42 -08:00
Danilo Peixoto 92ec632ad5 Fix Distributed Communication documentation (#1731) 
* Add missing `size()` method call for group
 2025-01-02 14:08:38 -08:00
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