L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Commit Graph

  • 39289ef025 [CUDA] Release build for cuda 13 (#2872) Awni Hannun 2025-12-04 21:42:26 -08:00
  • aefc9bd3f6 [CUDA] Faster general copy (#2873) Awni Hannun 2025-12-04 21:42:15 -08:00
  • 997cfc7699 Add a 2-pass col reduce for CUDA (#2863) Angelos Katharopoulos 2025-12-04 15:53:59 -08:00
  • 1fa8dc5797 Do a PyPi release for cuda on arm (#2866) Awni Hannun 2025-12-04 15:28:29 -08:00
  • a6d6717181 fix compile copying (#2871) Awni Hannun 2025-12-04 12:32:56 -08:00
  • 941cfe23d7 Layer norm throws on dimension mismatch (#2870) Awni Hannun 2025-12-04 11:21:05 -08:00
  • 9abb0b8123 Added support for pytree types that inherit from tuple and typing.namedtuple (#2845) romanoneg 2025-12-04 11:06:45 -08:00
  • 50d3914c67 Update gumbel function signature parameters (#2868) Tian En "TianHeng 2025-12-03 23:37:35 +00:00
  • cacbdbf995 Fix init from double (#2861) Awni Hannun 2025-12-03 06:08:11 -08:00
  • 193cdcd81a Fix graph updating (#2857) Awni Hannun 2025-12-02 17:12:24 -08:00
  • d8ceae7b77 Reduce JVP (#2854) Awni Hannun 2025-12-02 16:17:47 -08:00
  • eff0e31f00 Fix export scatters (#2852) Awni Hannun 2025-12-02 11:24:40 -08:00
  • 6c5785bc2f use thread local cpature mode (#2850) Awni Hannun 2025-12-01 19:02:47 -08:00
  • 8879ee00eb Support more Numpy interfaces for masked_scatter (#2832) CCYeh 2025-12-02 02:51:02 +01:00
  • 6e762fe2e2 [CUDA] Migrate conv code to new cuDNN APIs (#2847) Cheng 2025-12-02 07:55:43 +09:00
  • 2b95d0c270 [CUDA] Use cuDNN attention when T_q != T_kv (#2843) Cheng 2025-11-27 09:58:43 +09:00
  • b054838780 Added clarification to apply_fn parameter of apply_to_modules (#2831) Chaoran Yu 2025-11-26 15:40:56 -08:00
  • dd79d3c465 [CUDA] Faster rms norm for small dimension (#2838) Awni Hannun 2025-11-26 15:10:41 -08:00
  • 704fd1ae28 [CUDA] Support array mask in SDPA (#2822) Cheng 2025-11-26 11:08:58 +09:00
  • c9f4dc851f Merge build-cuda and build-linux actions (#2783) Cheng 2025-11-25 20:06:42 +09:00
  • f8bd675655 [CUDA] Output of SDPA should have same layout with inputs (#2826) Cheng 2025-11-25 15:22:58 +09:00
  • 23a9168d34 [CUDA] Add debug env to save cuda graphs to dot files (#2825) Cheng 2025-11-25 15:22:36 +09:00
  • bca205e287 [CUDA] Exit on crash and more helpful errors (#2830) Awni Hannun 2025-11-24 19:46:03 -08:00
  • 1d4eacb737 Fix mx.core.linspace type annotation (#2820) CCYeh 2025-11-24 23:15:08 +01:00
  • 8abd37ad05 Bump actions/checkout from 5 to 6 (#2828) dependabot[bot] 2025-11-24 06:04:46 -08:00
  • 3e05cea9f8 Force cudaGraphExec reinstantiation when clusters are used (#2813) Andrey Portnoy 2025-11-22 15:43:49 -05:00
  • 5b0f047226 Fix mx.core.load type annotation (#2819) CCYeh 2025-11-22 20:09:44 +01:00
  • 618c87af8c Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737) Harsh Sutaria 2025-11-22 09:51:36 -05:00
  • d5f61a93fa Fix typo: refs/head/main => refs/heads/main (#2818) Cheng 2025-11-22 09:43:35 +09:00
  • 4a09264236 Tolerance for some ops tests on cuda (#2815) Awni Hannun 2025-11-21 16:06:16 -08:00
  • 0dbc7e5bee Centralize NAX condition (#2811) Awni Hannun 2025-11-21 13:28:15 -08:00
  • 0d68efd461 patch bump for future version (#2804) Awni Hannun 2025-11-20 09:26:20 -08:00
  • f9e1a14135 [CUDA] Partly fix random for large sizes (#2798) Awni Hannun 2025-11-20 07:27:50 -08:00
  • d8e9ded928 Fix cuda allocator copy condition (#2800) Awni Hannun 2025-11-20 07:06:55 -08:00
  • 60939d010c Fix macos release target and linux arm release (#2802) Awni Hannun 2025-11-19 21:37:50 -08:00
  • fdcd2923fd patch + fix docs build (#2799) Awni Hannun 2025-11-19 16:16:26 -08:00
  • 54f1cc6e3e Add Neural Accelerator Support (#2772) v0.30.0 Jagrit Digani 2025-11-19 15:06:00 -08:00
  • b3825ac149 Add Masked Scatter (#2663) CCYeh 2025-11-19 23:53:32 +01:00
  • 7f4b7e553c version (#2797) Awni Hannun 2025-11-19 14:11:16 -08:00
  • ad16f41a7f Fix version tag (#2790) Awni Hannun 2025-11-19 08:55:57 -08:00
  • f46877bc08 more accurate rope fallback (#2792) Awni Hannun 2025-11-19 06:07:21 -08:00
  • 6f35017d1b [CUDA] cuDNN backward attention (#2762) Cheng 2025-11-19 08:13:50 +09:00
  • b167f0df1c build docs on linux (#2787) Awni Hannun 2025-11-18 08:01:03 -08:00
  • a9f0d6b160 Avoid duplicate CI runs when starting a PR from upstream branch (#2788) Cheng 2025-11-18 15:16:25 +09:00
  • 940f4c7818 Fix building with CUDA < 12.8 (#2782) Cheng 2025-11-18 12:55:19 +09:00
  • 35f81728f1 Remove unneeded tests in nightly build (#2786) Cheng 2025-11-18 08:09:58 +09:00
  • 4442ed86c1 Fix nightly build (#2785) Cheng 2025-11-18 08:07:51 +09:00
  • 698559c231 Test every commit in main branch (#2781) Cheng 2025-11-18 08:07:22 +09:00
  • ecc4879b07 Do not run CPU tests in CUDA builds (#2784) Cheng 2025-11-18 07:27:09 +09:00
  • 32b18d8b66 Use std::optional for mask_arr arg (#2763) Cheng 2025-11-17 10:43:33 +09:00
  • 472c43a0c8 Build and test with multiple CUDA versions (#2780) Cheng 2025-11-17 09:19:02 +09:00
  • b7214ff01e Remove pip cache in GitHub Actions (#2776) Cheng 2025-11-17 08:19:59 +09:00
  • 76414c8971 Run CI for pushes (#2777) Cheng 2025-11-17 07:19:01 +09:00
  • 49e4566df3 fix release 2 (#2767) Awni Hannun 2025-11-16 11:39:53 -08:00
  • aad49f932f [CUDA] Tune ops per buffer based on device (#2761) Awni Hannun 2025-11-16 06:29:49 -08:00
  • 86765cce34 Use ccache in GitHub Actions (#2773) Cheng 2025-11-16 07:58:14 +09:00
  • 1bedcbd556 Fix warnings with cmake 4.1 (#2774) Cheng 2025-11-16 07:12:47 +09:00
  • 9ac7dbe877 Fix MPI distributed tests with CUDA backend (#2775) Cheng 2025-11-16 07:12:18 +09:00
  • 1bf605d56d use arch specific targets when possible (#2771) Awni Hannun 2025-11-14 20:04:18 -08:00
  • 3c622ddd1d Separate test-linux from build-linux/cuda in GitHub Actions (#2765) Cheng 2025-11-15 11:14:09 +09:00
  • 27ff069175 Fix exporting with constants (#2769) Awni Hannun 2025-11-14 12:52:08 -08:00
  • 3b2ffcefc3 [CUDA] cuDNN forward attention (#2743) Cheng 2025-11-14 09:23:56 +09:00
  • b65f882df3 fix release (#2759) Awni Hannun 2025-11-13 15:34:01 -08:00
  • b704e9e77a [CUDA] Check CUDA error in synchronize (#2757) Cheng 2025-11-14 07:10:23 +09:00
  • 66519fb348 fix slice (#2758) Awni Hannun 2025-11-13 11:30:02 -08:00
  • 8973550ff3 export custom kernel (#2756) Awni Hannun 2025-11-13 11:29:50 -08:00
  • 3f866be665 minor debugging for publishing (#2739) Mike Drob 2025-11-12 08:33:39 -06:00
  • 23f81ed1c1 Linux on arm (#2751) Awni Hannun 2025-11-11 11:41:14 -08:00
  • 3fe2250c00 Fix irregular_strides benchmark shape type (#2754) wrmsr 2025-11-11 11:40:22 -08:00
  • 047114b988 remove circle (#2753) Awni Hannun 2025-11-11 11:39:47 -08:00
  • 9320eb89a8 Fix dequantize python sig (dtype default) (#2752) wrmsr 2025-11-11 09:55:24 -08:00
  • 75819d70ea patch bump (#2750) Awni Hannun 2025-11-11 08:49:14 -08:00
  • 60d80a3728 fix release builds (#2746) v0.29.4 Awni Hannun 2025-11-11 07:44:30 -08:00
  • eba6a9d163 Compatibility with pip-installed openmpi (#2741) Pedro Cuenca 2025-11-08 01:58:31 +01:00
  • be9e2aebd6 Shapeless support for zeros/ones_like (#2726) CCYeh 2025-11-07 04:12:20 +01:00
  • df58b4133a [CUDA] Reduce use of managed memory (#2725) Awni Hannun 2025-11-05 16:05:23 -08:00
  • 27778156dc Nccl reduce scatter, all gather (#2727) Anastasiia Filippova 2025-11-05 17:21:11 +01:00
  • 761f901a41 fix property name (#2736) Mike Drob 2025-11-05 06:31:56 -06:00
  • 6ece97f69b Make cpu binary_op easily accessible (#2733) Angelos Katharopoulos 2025-11-05 01:08:41 -08:00
  • d3bc6a9bff don't test when doing release (#2734) Awni Hannun 2025-11-04 15:54:23 -08:00
  • 26ceb507eb only build for macos 14 and up (#2731) Awni Hannun 2025-11-04 09:44:15 -08:00
  • 910b3e3299 skip self-hosted runners on forks (#2730) Mike Drob 2025-11-03 23:22:13 +01:00
  • 50fa315d18 Fix addmm with empty matrices and beta != 1.0 (#2715) Harsh Sutaria 2025-11-03 17:16:15 -05:00
  • 1ff2b713b6 Check isnan in maximum / minimum with CPU backend (#2652) AN Long 2025-11-04 01:51:14 +09:00
  • 50514a6146 Set up publishing to PyPI and Test-PyPI (#2721) Mike Drob 2025-11-03 16:20:11 +01:00
  • 93d76b0f30 Fix compile multi capture (#2678) Awni Hannun 2025-11-03 06:33:43 -08:00
  • 78678de0cd add null check -- the bundleIdentifier is optional (#2709) David Koski 2025-11-03 06:33:21 -08:00
  • ed9c6b1117 update: add linux fedora container CI - CPP build test only (#2722) Melissa Kilby 2025-11-03 06:33:00 -08:00
  • 24828b1b2f CMakeLists.txt update sign-warns Ronan Collobert 2025-10-31 16:55:04 -07:00
  • 9f649b5658 WIP (python) Ronan Collobert 2025-10-31 16:24:51 -07:00
  • 18aa921388 WIP Ronan Collobert 2025-10-31 16:24:35 -07:00
  • 8d13a0bc6b WIP (metal) Ronan Collobert 2025-10-31 16:24:21 -07:00
  • ac75c87fd7 WIP (cpu) Ronan Collobert 2025-10-31 16:24:09 -07:00
  • 7107802e09 WIP (examples) Ronan Collobert 2025-10-31 16:23:51 -07:00
  • c5913131cf WIP (distributed) Ronan Collobert 2025-10-31 13:32:56 -07:00
  • 19ab7911f6 WIP (cuda) Ronan Collobert 2025-10-31 13:32:43 -07:00
  • 4a1b1796b7 WIP (io) Ronan Collobert 2025-10-31 13:20:47 -07:00
  • b48d298205 WIP (distributed) Ronan Collobert 2025-10-31 13:20:09 -07:00
  • 8277e71ea9 WIP (gpu) Ronan Collobert 2025-10-31 13:19:54 -07:00
  • b0d985416a fix arg_reduce Ronan Collobert 2025-10-31 13:13:15 -07:00