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

Commit Graph

  • eb64c60144 docs Awni Hannun 2023-12-07 08:00:41 -08:00
  • 86bd60c849 docs Awni Hannun 2023-12-06 08:13:20 -08:00
  • 7a235ce49e docs Awni Hannun 2023-12-05 14:18:20 -08:00
  • 13e8d87a88 docs Awni Hannun 2023-12-05 12:10:03 -08:00
  • e09a97e24e docs Awni Hannun 2023-11-30 11:55:34 -08:00
  • aa0e5e9049 docs Awni Hannun 2023-11-30 11:55:10 -08:00
  • 1c70552af7 docs Awni Hannun 2023-11-29 12:46:18 -08:00
  • a372a3844d docs Awni Hannun 2023-11-29 12:41:56 -08:00
  • 9f0ba3ddf1 patch bump (#2680) Awni Hannun 2025-10-17 12:12:07 -07:00
  • 4bce5f9b2d suppress gcc 10.1 warnings (#2679) v0.29.3 Awni Hannun 2025-10-17 12:09:21 -07:00
  • e9eab527eb Nccl timeout (#2673) Anastasiia Filippova 2025-10-14 21:29:54 +02:00
  • 36ca62dba8 remove unused unary file (#2672) Awni Hannun 2025-10-13 19:36:26 -07:00
  • 9cbb1b0148 Modified sort behavior when running CPU or Metal to match NumPy/JAX (#2667) Manuel Villanueva 2025-10-13 16:36:45 -05:00
  • 9bfc476d72 Normalize README bullet formatting (#2671) Fabrizio Milo 2025-10-13 12:13:30 -07:00
  • 25e2356316 speed up scalars (#2669) Awni Hannun 2025-10-13 12:10:15 -07:00
  • 226a1d24e0 Debug cuda conv (#2662) Awni Hannun 2025-10-10 16:12:47 -07:00
  • 630350ad3e Precise sigmoid (#2659) Awni Hannun 2025-10-10 10:05:23 -07:00
  • 380aeb58ae enable admm low-precision cpu (#2661) Awni Hannun 2025-10-10 09:50:54 -07:00
  • f37389d100 bump patch (#2658) Awni Hannun 2025-10-10 08:36:41 -07:00
  • e89e8b4272 Export with callback (#2612) Awni Hannun 2025-10-08 19:24:33 -07:00
  • 85a8824a8c Fix cumulative operations when axis=None (#2653) AN Long 2025-10-09 07:25:38 +09:00
  • f5d4397e5c Fix fast synch when fence is waited before a command buffer is created (#2657) Awni Hannun 2025-10-08 11:23:46 -07:00
  • 343e33b6d5 fix all_gather vjp (#2654) Awni Hannun 2025-10-07 06:05:23 -07:00
  • 0073096dd1 Split name into directories for cuda jit (#2656) Angelos Katharopoulos 2025-10-07 01:52:58 -07:00
  • e3d004fed9 Fix and refactor row-reduce (#2650) Angelos Katharopoulos 2025-10-07 01:51:08 -07:00
  • a393435d28 Speed up compile for node with many parents (#2649) Awni Hannun 2025-10-03 19:30:36 -07:00
  • a7a94b29d7 Fix compile when outputs change (#2648) Awni Hannun 2025-10-03 08:40:57 -07:00
  • 22a5da76c8 Faster complex matmul (#2571) Daniel Yeh 2025-10-03 08:33:15 +02:00
  • 287c63a093 Configure CMake to export compile_commands.json (#2645) Andrey Portnoy 2025-10-02 18:40:32 -04:00
  • 1c9ae1eaa1 cuda fix flaky test (#2646) Awni Hannun 2025-10-02 15:40:04 -07:00
  • c2c3e0b0a2 [CUDA] Add a small column specialization to reduce (#2642) Angelos Katharopoulos 2025-10-02 14:41:05 -07:00
  • b0cc71ae71 Faster triu, tril, where with scalar (#2644) Awni Hannun 2025-10-02 12:21:27 -07:00
  • e88f2d4a8e fix cross entropy axis param (#2641) Awni Hannun 2025-10-01 16:49:55 -07:00
  • 9cee557423 Fix status message (#2638) Angelos Katharopoulos 2025-10-01 16:43:45 -07:00
  • bbf1423953 wait for tasks in cuda (#2636) Awni Hannun 2025-09-30 16:08:46 -07:00
  • eb24267b56 Compile now can attach arbitrary data to an entry (#2634) Angelos Katharopoulos 2025-09-30 13:33:27 -07:00
  • dc371ae7a5 fix for max block dim (#2631) Awni Hannun 2025-09-29 08:59:25 -07:00
  • e76a8dd5c5 Fix incorrect path and typos (#2630) AN Long 2025-09-28 22:03:04 +09:00
  • b466dea982 [CUDA] Make CudaEvent work with multi-device (#2614) Cheng 2025-09-27 11:27:17 +09:00
  • 7a6adda1e6 Bump the version (#2627) v0.29.2 Angelos Katharopoulos 2025-09-26 15:15:28 -07:00
  • 1a9f820af6 Compiled should not end in broadcast (#2622) Angelos Katharopoulos 2025-09-26 13:36:09 -07:00
  • d4f4ff3c5e Allow None input to compiled functions (#2621) Awni Hannun 2025-09-25 08:42:23 -07:00
  • 7c7e48dbd1 New tuning for small K gemv (#2620) Jagrit Digani 2025-09-23 12:28:35 -07:00
  • fbbf3b9b3e Support pickling array for bfloat16 (#2586) Daniel Yeh 2025-09-23 05:12:15 +02:00
  • bf01ad9367 fix (#2613) Daniel Yeh 2025-09-23 05:12:04 +02:00
  • ae438d05fa [CUDA] Recycle CUDA events (#2604) Cheng 2025-09-23 10:42:03 +09:00
  • 711a645807 avoid producing NaN in attention (#2608) Awni Hannun 2025-09-22 13:10:43 -07:00
  • aa9d44b3d4 implement Convolution::output_shape (#2601) Josh Bleecher Snyder 2025-09-22 10:09:45 -07:00
  • ec2ab42888 Lower sorted QMM gather threshold (#2609) Awni Hannun 2025-09-19 18:22:55 -07:00
  • 787c0d90cd Detect cache thrashing in LRUCache (#2600) Cheng 2025-09-19 09:12:14 +09:00
  • e8b604a6a3 fix: library loading for swift dynamic frameworks (#2568) Oleksandr Bilous 2025-09-18 23:54:59 +03:00
  • 50cc09887f expose depends (#2606) Awni Hannun 2025-09-18 10:06:15 -07:00
  • 3f730e77aa Update export function example for array input (#2598) Umberto Mignozzetti 2025-09-16 14:38:05 -07:00
  • caecbe876a no copy batch rope (#2595) Awni Hannun 2025-09-15 14:23:48 -07:00
  • 8afb6d62f2 Fix typo in average_gradients function call (#2594) Umberto Mignozzetti 2025-09-15 11:29:21 -07:00
  • 6ccfa603cd fix metal scan (#2591) Awni Hannun 2025-09-15 11:01:57 -07:00
  • 36cad99a11 Refactor code examples to use 'gelu' (#2592) Umberto Mignozzetti 2025-09-15 09:47:02 -07:00
  • ee18e1cbf0 patch bump (#2588) v0.29.1 Awni Hannun 2025-09-11 17:10:09 -07:00
  • af120c2bc0 set nccl ABI version (#2587) Awni Hannun 2025-09-11 16:55:53 -07:00
  • 6a3acf2301 [CUDA] Set bias as input when using bias epilogue (#2584) Cheng 2025-09-11 15:31:09 +09:00
  • d6977f2a57 Add sdpa with sinks (#2558) Awni Hannun 2025-09-10 14:53:00 -07:00
  • db5443e831 Adding Relu2 (#2582) Gökdeniz Gülmez 2025-09-10 16:24:30 +02:00
  • 52b8384d10 Fix flaky addmm tests (#2581) Cheng 2025-09-10 14:22:22 +09:00
  • 44cc5da4bc [CUDA] Fix alpha not respected when using bias epilogue (#2578) Cheng 2025-09-10 09:08:01 +09:00
  • dde3682b69 [CUDA] Use GEMM with epilogue instead of AddMM (#2569) Cheng 2025-09-09 13:18:49 +09:00
  • 17310d91a6 Add batch offsets for mx.fast.rope (#2564) Awni Hannun 2025-09-08 17:35:07 -07:00
  • b194d65a6a Some tweaks in cmake files (#2574) Cheng 2025-09-09 08:27:18 +09:00
  • a44b27f5f8 Fix a few ccache cache miss (#2573) Cheng 2025-09-09 07:41:05 +09:00
  • e5a33f2223 faster depthwise 1D conv (#2567) Awni Hannun 2025-09-08 11:37:23 -07:00
  • c1e3340b23 Set ccache size before building (#2570) Cheng 2025-09-07 09:00:31 +09:00
  • 8f163a367d typing: add type hints to mlx.core.array, linalg, distributed, and random (#2565) XXXXRT666 2025-09-05 00:08:11 +08:00
  • 89a3df9014 Fixed several type annotations in the MLX stubs which degraded to Unknown/Any (#2560) Manuel Villanueva 2025-09-03 14:52:08 -05:00
  • c5d2937aa5 chore: Update Docs With Slice Copy Example (#2559) Krishi Saripalli 2025-09-02 22:07:02 -07:00
  • b61a65e313 fix copies in sdpa (#2563) Awni Hannun 2025-09-02 11:00:36 -07:00
  • 04cbb4191c Fix dequantize python sig (#2562) wrmsr 2025-09-01 11:50:20 -07:00
  • c5460762e7 Fix AdamW weight_decay default value in docstring (#2557) Artur Antonov 2025-09-01 07:29:30 +03:00
  • 8ce49cd39e fix quantized vjp for mxfp4 (#2555) v0.29.0 Awni Hannun 2025-08-29 10:06:15 -07:00
  • 9c68b50853 version bump (#2554) Awni Hannun 2025-08-29 06:54:17 -07:00
  • 111f1e71af Faster contiguous gather for indices in the first axis (#2552) Awni Hannun 2025-08-28 21:26:30 -07:00
  • 827003d568 fix METAL quantization in JIT (#2553) Awni Hannun 2025-08-28 18:26:25 -07:00
  • d363a76aa4 Bump xcode in circle (#2551) Awni Hannun 2025-08-28 13:13:34 -07:00
  • 70560b6bd5 Add mode parameter for quantization (#2499) Awni Hannun 2025-08-28 06:45:26 -07:00
  • 7ef8a6f2d5 [CUDA] fix sort (#2550) Awni Hannun 2025-08-27 19:48:43 -07:00
  • 31c6f6e33f [CUDA] Use ConcurrentContext in concatenate_gpu (#2549) Cheng 2025-08-28 09:30:08 +09:00
  • 584d48458e link with nccl (#2546) Awni Hannun 2025-08-27 10:01:07 -07:00
  • 5cf984ca87 Separate cpu compilation cache by versions (#2548) Cheng 2025-08-27 11:25:15 +09:00
  • a9bac3d9e5 Run CPP tests for CUDA build in CI (#2544) Cheng 2025-08-27 08:06:46 +09:00
  • 5458d43247 add load with path tests (#2543) Awni Hannun 2025-08-26 14:24:47 -07:00
  • a4dba65220 Enable cuda graph toggle (#2545) Awni Hannun 2025-08-26 12:50:38 -07:00
  • 3dcb286baf Remove stream from average grads so it uses default (#2532) Awni Hannun 2025-08-25 15:56:29 -07:00
  • 4822c3dbe9 [CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533) Cheng 2025-08-26 07:31:39 +09:00
  • 2ca75bb529 Remove nccl install in release (#2542) Awni Hannun 2025-08-25 15:20:18 -07:00
  • db14e29a0b allow pathlib.Path to save/load functions (#2541) Awni Hannun 2025-08-25 14:58:49 -07:00
  • d2f540f4e0 Use nccl header only when nccl is not present (#2539) Awni Hannun 2025-08-25 14:17:25 -07:00
  • 333ffea273 [CUDA] Remove thrust in arange (#2535) Cheng 2025-08-24 16:22:36 +09:00
  • f55b6f1f2f Enable COMPILE_WARNING_AS_ERROR for linux builds in CI (#2534) Cheng 2025-08-24 15:33:08 +09:00
  • 30561229c7 Fix allocation bug in NCCL (#2530) Awni Hannun 2025-08-22 14:39:43 -07:00
  • 068a4612e9 nccl default for backend=any (#2528) Awni Hannun 2025-08-22 12:24:27 -07:00
  • 5722c147de [CUDA] Update calls to cudaMemAdvise and cudaGraphAddDependencies for CUDA 13 (#2525) Andrey Portnoy 2025-08-21 22:57:20 -04:00
  • f6819a1f26 Fix warning 186-D from nvcc (#2527) Cheng 2025-08-22 10:29:55 +09:00