Commit Graph

  • 481349495b GPU Hadamard for large N (#1879) Angelos Katharopoulos 2025-02-18 13:43:09 -0800
  • 9daa6b003f
    fix shapeless export (#2148) Awni Hannun 2025-05-01 15:02:02 -0700
  • 023b718877 fix shapeless export Awni Hannun 2025-05-01 14:21:29 -0700
  • a3a632d567
    Fix the launcher when ran locally (#2147) Angelos Katharopoulos 2025-05-01 12:56:09 -0700
  • 53d446da53 Fix the launcher when ran locally Angelos Katharopoulos 2025-05-01 11:46:24 -0700
  • 28902ece4e updated, simplified mutex for thread safety Andrew Sweet 2025-04-30 16:17:12 -0700
  • e496c5a4b4
    fix integer overflow in qmm (#2143) Awni Hannun 2025-04-30 09:28:56 -0700
  • ea890d8710
    Remove metal-only tests (#2139) Cheng 2025-05-01 01:08:39 +0900
  • aa5d84f102
    Allow quant layer to be unfrozen (#2142) Awni Hannun 2025-04-30 09:08:29 -0700
  • f1606486d2
    Generalize gpu backend (#2138) Awni Hannun 2025-04-30 09:08:17 -0700
  • 45df803538 fix integer overflow in qmm Awni Hannun 2025-04-30 09:04:12 -0700
  • 81e8aed49d Allow quant layer to be unfrozen Awni Hannun 2025-04-30 07:13:08 -0700
  • 08ef9408b5 Remove metal-only tests Cheng 2025-04-30 15:06:00 +0900
  • 87720a8908
    Fix building with uv (#2141) Cheng 2025-04-30 22:04:07 +0900
  • ae587aae0c Fix building with uv Cheng 2025-04-30 21:12:44 +0900
  • bb6565ef14
    add fftshift and ifftshift fft helpers (#2135) Aashiq Dheeraj 2025-04-30 01:13:45 -0400
  • 6389b9d37a fix fp error in roll + add test Aashiq Dheeraj 2025-04-29 23:04:09 -0400
  • 9cfe0b1533 axes have to be iterable Aashiq Dheeraj 2025-04-29 20:59:05 -0400
  • 7c9746a0bc address comments Aashiq Dheeraj 2025-04-29 20:42:27 -0400
  • be1d68e1ad generalize gpu backend Awni Hannun 2025-04-29 16:05:41 -0700
  • 6151995219 fix no_gpu build Awni Hannun 2025-04-29 15:29:08 -0700
  • f6a23c44e3 fix no_gpu build Awni Hannun 2025-04-29 15:27:59 -0700
  • 949f0163dd generalize gpu backend Awni Hannun 2025-04-29 15:21:03 -0700
  • 7bb063bcb3
    Enable vjp for quantized scale and bias (#2129) Awni Hannun 2025-04-29 13:03:09 -0700
  • 89d91a7c00 higher tol Awni Hannun 2025-04-29 12:41:07 -0700
  • b36dd472bb
    return library if it is successfully loaded (#2131) Alex Chi Z. 2025-04-29 10:30:36 -0400
  • 167b759a38
    Fix typos (#2136) hdeng-apple 2025-04-29 22:26:05 +0800
  • dde2c66832 Fix typos Hanyu Deng 2025-04-29 13:50:17 +0800
  • 00e43d18ed add fftshift and ifftshift fft helpers Aashiq Dheeraj 2025-04-29 00:28:01 -0400
  • 9092ff772e
    return library if it is successfully loaded Alex Chi Z. 2025-04-27 23:57:59 -0400
  • c384e4cbc6 Enable vjp for quantized scale and bias Awni Hannun 2025-04-26 09:27:01 -0700
  • 0244a15318 Added jacfwd, jacrev and hessian paramthakkar123 2025-04-26 20:27:51 +0530
  • 998404ada4 Get trellis to run trellis-quants Awni Hannun 2025-04-26 07:02:20 -0700
  • 99b9868859
    Clarify dimension notation in conv1d, conv2d, and conv3d docstrings (#2123) charan-003 2025-04-25 13:18:30 -0600
  • 09a74bc516 Updating transposed convs in conv1d, conv2d, and conv3d Sai Charan Arvapally 2025-04-25 12:38:25 -0600
  • 6b2d5448f2
    Fix the error message in mx.right_shift and mx.left_shift (#2121) 1ndig0 2025-04-26 00:14:28 +0800
  • 0f18a974a3 simplify Awni Hannun 2025-04-25 06:30:52 -0700
  • 268314cc2f Clarify dimension notation in conv1d, conv2d, and conv3d docstrings Sai Charan Arvapally 2025-04-25 02:01:52 -0600
  • 803a417ed7 update right_shift and lef_shift Redmept1on 2025-04-25 11:29:25 +0800
  • eaf709b83e
    patch (#2119) v0.25.1 Awni Hannun 2025-04-24 16:11:07 -0700
  • 6c3eff6691 patch Awni Hannun 2025-04-24 15:13:54 -0700
  • f0e70afff0
    Fix swift pm load (#2117) Angelos Katharopoulos 2025-04-24 10:58:29 -0700
  • 7906590c0e Another fix Angelos Katharopoulos 2025-04-24 10:09:33 -0700
  • 9bde6a95eb Fix swift pm load Angelos Katharopoulos 2025-04-24 09:51:40 -0700
  • 86984cad68
    Remove static initializers (#2059) hdeng-apple 2025-04-24 21:14:49 +0800
  • f034b51e38 Remove duplicate private section of thread_pool() Hanyu Deng 2025-04-23 11:00:57 +0800
  • 27ceff33d3 Revert changes in pocketfft.h Hanyu Deng 2025-04-10 22:37:36 +0800
  • b7ce3c947c Remove static initializer of CompilerCache cache Hanyu Deng 2025-04-10 15:01:17 +0800
  • 83d320c2ea Remove static initializer InTracing::trace_stack Hanyu Deng 2025-04-10 12:00:52 +0800
  • aaf261d128 Remove static initializers in device.cpp, load.cpp, pocketfft.h Hanyu Deng 2025-04-10 11:43:51 +0800
  • fbc89e3ced
    fix pinv (#2110) Awni Hannun 2025-04-23 13:08:28 -0700
  • 38c1e720c2
    Search mlx.metallib in macOS framework "Resources" dir (#2061) hdeng-apple 2025-04-24 00:53:13 +0800
  • 600e87e03c
    Added output_padding parameters in conv_transpose (#2092) Param Thakkar 2025-04-23 21:56:33 +0530
  • f7f323f6ae Added stft and istft in the header paramthakkar123 2025-04-23 19:19:04 +0530
  • a7a6c49909 Add jacfwd paramthakkar123 2025-04-23 18:47:19 +0530
  • 85ee89fcf0 fix pinv Awni Hannun 2025-04-22 20:01:58 -0700
  • 3836445241
    Add broadcast_shapes in python API (#2091) Hyunsung Lee 2025-04-23 10:57:39 +0900
  • 1d2c9d6a07
    Complex scan (#2094) Yury Popov 2025-04-23 04:56:28 +0300
  • 3f202e568a Fix cpp tests Angelos Katharopoulos 2025-04-22 18:51:32 -0700
  • c47ae01213 Check in Resources as well Angelos Katharopoulos 2025-04-22 16:18:17 -0700
  • b16a1d50b2 Revert "Search mlx.metallib in macOS framework" Angelos Katharopoulos 2025-04-22 16:19:05 -0700
  • f360823cee Search mlx.metallib in macOS framework Hanyu Deng 2025-04-10 18:17:12 +0800
  • 7a41a7051b Fix conv_general and docstrings Angelos Katharopoulos 2025-04-22 15:41:41 -0700
  • a963a15b8d
    Merge branch 'main' into stft Param Thakkar 2025-04-23 00:02:52 +0530
  • e8ac6bd2f5
    irfft throws instead of segfaults on scalars (#2109) Awni Hannun 2025-04-22 10:25:55 -0700
  • 05518cbbad irfft throws instead of segfaults on scalars Awni Hannun 2025-04-22 06:28:35 -0700
  • f2b5ba49af Added output_padding to nn.layers as well as tests for the same paramthakkar123 2025-04-22 18:55:06 +0530
  • e857189b2a
    remove unnecessary tempspec Yury Popov 2025-04-22 10:37:16 +0300
  • 11f73d6e89 Double buffer keys for vector sdpa sdpa-test Angelos Katharopoulos 2025-04-22 00:19:11 -0700
  • 605c6bc551 nit Hyunsung Lee 2025-04-22 05:55:41 +0900
  • 5e8712cf6f Fix upon review Update python/src/ops.cpp Hyunsung Lee 2025-04-22 05:47:07 +0900
  • fdadc4f22c
    Add more complex unary ops (#2101) Awni Hannun 2025-04-21 13:04:54 -0700
  • 79b527f45f
    conv vmap (#2102) Awni Hannun 2025-04-21 13:04:39 -0700
  • d3d575cce7
    Merge branch 'ml-explore:main' into adding-Muon-optimizer Gökdeniz Gülmez 2025-04-21 20:27:33 +0200
  • db6b0203dc conv vmap Awni Hannun 2025-04-18 17:40:41 -0700
  • dc4eada7f0
    Use unordered map for kwargs in export/import (#2087) Awni Hannun 2025-04-21 07:17:22 -0700
  • 70ebc3b598
    Return const ref in array::data_shared_ptr (#2100) Cheng 2025-04-21 22:17:09 +0800
  • 6df48fe34a Add more complex unary ops Awni Hannun 2025-04-21 07:15:12 -0700
  • ed4b77d7ee comment Awni Hannun 2025-04-21 06:11:34 -0700
  • f59ddb399c Return const ref in array::data_shared_ptr Cheng 2025-04-21 19:41:01 +0800
  • 3787167fb2 nit Hyunsung Lee 2025-04-20 22:47:37 +0900
  • ce99b8848b
    tests: complex logaddexp / logcumsumexp Yury Popov 2025-04-20 16:43:03 +0300
  • ed777f2ec6 fix upon review Hyunsung Lee 2025-04-20 22:38:39 +0900
  • 26d0e56e9f
    tests: add complex scans Yury Popov 2025-04-20 15:52:39 +0300
  • a19b21ad22
    tests: add complex log1p Yury Popov 2025-04-20 15:49:21 +0300
  • c3c2bdb194
    metal: add complex logaddexp Yury Popov 2025-04-20 02:06:54 +0300
  • 0cdccd54ae
    metal: add complex log1p Yury Popov 2025-04-20 01:28:28 +0300
  • e444e2cbc0
    cpu: allow complex logaddexp / atan2 Yury Popov 2025-04-20 02:06:28 +0300
  • f5dc7eaf6a nit Hyunsung Lee 2025-04-20 13:01:36 +0900
  • 92444f393b nit Hyunsung Lee 2025-04-20 12:58:26 +0900
  • a7a96b0ad6 fix to use c++ api Hyunsung Lee 2025-04-20 12:55:58 +0900
  • 49afaf8f44
    Update python/src/CMakeLists.txt Hyunsung Lee 2025-04-20 12:05:18 +0900
  • c7e326bb44
    cpu: complex scan dispatch Yury Popov 2025-04-20 00:16:08 +0300
  • 38f593026c
    cpu: add complex log1p Yury Popov 2025-04-20 00:15:48 +0300
  • 59b934dcf9 Added conv_transpose fixes paramthakkar123 2025-04-19 20:59:53 +0530
  • b13f2aed16
    Introduce macros for dispatching dynamic dtypes as static types (#2073) Cheng 2025-04-19 21:16:30 +0800
  • 876c1986e4 fix copyright text Hyunsung Lee 2025-04-19 15:43:58 +0900
  • 5f62e209f4 Add broadcast_shapes in python API Hyunsung Lee 2025-04-19 15:37:19 +0900
  • 8ac0a20250 SDPA fixes paramthakkar123 2025-04-19 08:15:56 +0530
  • 36b472dc6c use unordered map for kwargs in export/import Awni Hannun 2025-04-18 18:07:39 -0700