mlx 0.30.7.2 → 0.30.7.6

metadata

@@ -1,15 +1,85 @@
 --- !ruby/object:Gem::Specification
 name: mlx
 version: !ruby/object:Gem::Version
-  version: 0.30.7.2
+  version: 0.30.7.6
 platform: ruby
 authors:
 - MLX Contributors
 - Aleksey Skryl
 bindir: bin
 cert_chain: []
-date: 2026-02-14 00:00:00.000000000 Z
-dependencies: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ostruct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: benchmark
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A Ruby wrapper for the native MLX machine learning runtime.
 email:
 - mlx@group.apple.com
@@ -19,23 +89,36 @@ extensions:
 - ext/mlx/extconf.rb
 extra_rdoc_files: []
 files:
+- ext/mlx-onnx/native.cpp
+- ext/mlx-onnx/native.hpp
 - ext/mlx/extconf.rb
 - ext/mlx/native.cpp
+- lib/mlx-onnx/webgpu_harness.rb
 - lib/mlx.rb
 - lib/mlx/core.rb
 - lib/mlx/distributed_utils/common.rb
 - lib/mlx/distributed_utils/config.rb
 - lib/mlx/distributed_utils/launch.rb
 - lib/mlx/dsl.rb
+- lib/mlx/dsl/attention.rb
 - lib/mlx/dsl/builder.rb
+- lib/mlx/dsl/config_schema.rb
 - lib/mlx/dsl/data_pipeline.rb
 - lib/mlx/dsl/experiment.rb
+- lib/mlx/dsl/generate.rb
 - lib/mlx/dsl/graph_modules.rb
+- lib/mlx/dsl/kv_cache.rb
+- lib/mlx/dsl/masks.rb
 - lib/mlx/dsl/model.rb
 - lib/mlx/dsl/model_mixin.rb
+- lib/mlx/dsl/positions.rb
+- lib/mlx/dsl/run_stack.rb
 - lib/mlx/dsl/split_plan.rb
+- lib/mlx/dsl/tensor.rb
 - lib/mlx/dsl/train_step.rb
 - lib/mlx/dsl/trainer.rb
+- lib/mlx/dsl/transformer_block.rb
+- lib/mlx/dsl/weight_map.rb
 - lib/mlx/extension.rb
 - lib/mlx/nn.rb
 - lib/mlx/nn/base.rb
@@ -59,573 +142,588 @@ files:
 - lib/mlx/nn/layers/upsample.rb
 - lib/mlx/nn/losses.rb
 - lib/mlx/nn/utils.rb
+- lib/mlx/onnx.rb
 - lib/mlx/optimizers.rb
 - lib/mlx/optimizers/optimizers.rb
 - lib/mlx/optimizers/schedulers.rb
 - lib/mlx/utils.rb
 - lib/mlx/version.rb
-- mlx/CMakeLists.txt
-- mlx/cmake/FindCUDNN.cmake
-- mlx/cmake/FindNCCL.cmake
-- mlx/cmake/Findnvpl.cmake
-- mlx/cmake/extension.cmake
-- mlx/mlx.pc.in
-- mlx/mlx/3rdparty/.clang-format
-- mlx/mlx/3rdparty/pocketfft.h
-- mlx/mlx/CMakeLists.txt
-- mlx/mlx/allocator.h
-- mlx/mlx/api.h
-- mlx/mlx/array.cpp
-- mlx/mlx/array.h
-- mlx/mlx/backend/common/CMakeLists.txt
-- mlx/mlx/backend/common/binary.h
-- mlx/mlx/backend/common/broadcasting.cpp
-- mlx/mlx/backend/common/broadcasting.h
-- mlx/mlx/backend/common/buffer_cache.h
-- mlx/mlx/backend/common/common.cpp
-- mlx/mlx/backend/common/compiled.cpp
-- mlx/mlx/backend/common/compiled.h
-- mlx/mlx/backend/common/copy.h
-- mlx/mlx/backend/common/hadamard.h
-- mlx/mlx/backend/common/load.cpp
-- mlx/mlx/backend/common/matmul.h
-- mlx/mlx/backend/common/reduce.cpp
-- mlx/mlx/backend/common/reduce.h
-- mlx/mlx/backend/common/slicing.cpp
-- mlx/mlx/backend/common/slicing.h
-- mlx/mlx/backend/common/ternary.h
-- mlx/mlx/backend/common/unary.h
-- mlx/mlx/backend/common/utils.cpp
-- mlx/mlx/backend/common/utils.h
-- mlx/mlx/backend/cpu/CMakeLists.txt
-- mlx/mlx/backend/cpu/arange.h
-- mlx/mlx/backend/cpu/arg_reduce.cpp
-- mlx/mlx/backend/cpu/binary.cpp
-- mlx/mlx/backend/cpu/binary.h
-- mlx/mlx/backend/cpu/binary_ops.h
-- mlx/mlx/backend/cpu/binary_two.h
-- mlx/mlx/backend/cpu/cholesky.cpp
-- mlx/mlx/backend/cpu/compiled.cpp
-- mlx/mlx/backend/cpu/compiled_preamble.h
-- mlx/mlx/backend/cpu/conv.cpp
-- mlx/mlx/backend/cpu/copy.cpp
-- mlx/mlx/backend/cpu/copy.h
-- mlx/mlx/backend/cpu/device_info.cpp
-- mlx/mlx/backend/cpu/device_info.h
-- mlx/mlx/backend/cpu/distributed.cpp
-- mlx/mlx/backend/cpu/eig.cpp
-- mlx/mlx/backend/cpu/eigh.cpp
-- mlx/mlx/backend/cpu/encoder.cpp
-- mlx/mlx/backend/cpu/encoder.h
-- mlx/mlx/backend/cpu/eval.cpp
-- mlx/mlx/backend/cpu/eval.h
-- mlx/mlx/backend/cpu/fft.cpp
-- mlx/mlx/backend/cpu/gemm.h
-- mlx/mlx/backend/cpu/gemms/bnns.cpp
-- mlx/mlx/backend/cpu/gemms/cblas.cpp
-- mlx/mlx/backend/cpu/gemms/simd_bf16.cpp
-- mlx/mlx/backend/cpu/gemms/simd_fp16.cpp
-- mlx/mlx/backend/cpu/gemms/simd_gemm.h
-- mlx/mlx/backend/cpu/hadamard.cpp
-- mlx/mlx/backend/cpu/indexing.cpp
-- mlx/mlx/backend/cpu/inverse.cpp
-- mlx/mlx/backend/cpu/jit_compiler.cpp
-- mlx/mlx/backend/cpu/jit_compiler.h
-- mlx/mlx/backend/cpu/lapack.h
-- mlx/mlx/backend/cpu/logsumexp.cpp
-- mlx/mlx/backend/cpu/luf.cpp
-- mlx/mlx/backend/cpu/make_compiled_preamble.ps1
-- mlx/mlx/backend/cpu/make_compiled_preamble.sh
-- mlx/mlx/backend/cpu/masked_mm.cpp
-- mlx/mlx/backend/cpu/matmul.cpp
-- mlx/mlx/backend/cpu/primitives.cpp
-- mlx/mlx/backend/cpu/qrf.cpp
-- mlx/mlx/backend/cpu/quantized.cpp
-- mlx/mlx/backend/cpu/reduce.cpp
-- mlx/mlx/backend/cpu/scan.cpp
-- mlx/mlx/backend/cpu/select.cpp
-- mlx/mlx/backend/cpu/simd/accelerate_fp16_simd.h
-- mlx/mlx/backend/cpu/simd/accelerate_simd.h
-- mlx/mlx/backend/cpu/simd/base_simd.h
-- mlx/mlx/backend/cpu/simd/math.h
-- mlx/mlx/backend/cpu/simd/neon_fp16_simd.h
-- mlx/mlx/backend/cpu/simd/simd.h
-- mlx/mlx/backend/cpu/simd/type.h
-- mlx/mlx/backend/cpu/slicing.h
-- mlx/mlx/backend/cpu/softmax.cpp
-- mlx/mlx/backend/cpu/sort.cpp
-- mlx/mlx/backend/cpu/svd.cpp
-- mlx/mlx/backend/cpu/ternary.h
-- mlx/mlx/backend/cpu/threefry.cpp
-- mlx/mlx/backend/cpu/threefry.h
-- mlx/mlx/backend/cpu/unary.cpp
-- mlx/mlx/backend/cpu/unary.h
-- mlx/mlx/backend/cpu/unary_ops.h
-- mlx/mlx/backend/cuda/CMakeLists.txt
-- mlx/mlx/backend/cuda/allocator.cpp
-- mlx/mlx/backend/cuda/allocator.h
-- mlx/mlx/backend/cuda/arange.cu
-- mlx/mlx/backend/cuda/arg_reduce.cu
-- mlx/mlx/backend/cuda/bin2h.cmake
-- mlx/mlx/backend/cuda/binary/CMakeLists.txt
-- mlx/mlx/backend/cuda/binary/add.cu
-- mlx/mlx/backend/cuda/binary/arctan2.cu
-- mlx/mlx/backend/cuda/binary/binary.cuh
-- mlx/mlx/backend/cuda/binary/bitwise_binary.cu
-- mlx/mlx/backend/cuda/binary/divide.cu
-- mlx/mlx/backend/cuda/binary/equal.cu
-- mlx/mlx/backend/cuda/binary/greater.cu
-- mlx/mlx/backend/cuda/binary/greater_equal.cu
-- mlx/mlx/backend/cuda/binary/less.cu
-- mlx/mlx/backend/cuda/binary/less_equal.cu
-- mlx/mlx/backend/cuda/binary/log_add_exp.cu
-- mlx/mlx/backend/cuda/binary/logical_and.cu
-- mlx/mlx/backend/cuda/binary/logical_or.cu
-- mlx/mlx/backend/cuda/binary/maximum.cu
-- mlx/mlx/backend/cuda/binary/minimum.cu
-- mlx/mlx/backend/cuda/binary/multiply.cu
-- mlx/mlx/backend/cuda/binary/not_equal.cu
-- mlx/mlx/backend/cuda/binary/power.cu
-- mlx/mlx/backend/cuda/binary/remainder.cu
-- mlx/mlx/backend/cuda/binary/subtract.cu
-- mlx/mlx/backend/cuda/binary_two.cu
-- mlx/mlx/backend/cuda/compiled.cpp
-- mlx/mlx/backend/cuda/conv.cpp
-- mlx/mlx/backend/cuda/conv/conv.h
-- mlx/mlx/backend/cuda/conv/gemm_conv.cu
-- mlx/mlx/backend/cuda/conv/gemm_grouped_conv.cu
-- mlx/mlx/backend/cuda/copy.cu
-- mlx/mlx/backend/cuda/copy/copy.cuh
-- mlx/mlx/backend/cuda/copy/copy_contiguous.cu
-- mlx/mlx/backend/cuda/copy/copy_general.cu
-- mlx/mlx/backend/cuda/copy/copy_general_dynamic.cu
-- mlx/mlx/backend/cuda/copy/copy_general_input.cu
-- mlx/mlx/backend/cuda/cublas_utils.cpp
-- mlx/mlx/backend/cuda/cublas_utils.h
-- mlx/mlx/backend/cuda/cuda.h
-- mlx/mlx/backend/cuda/cuda_utils.h
-- mlx/mlx/backend/cuda/cudnn_utils.cpp
-- mlx/mlx/backend/cuda/cudnn_utils.h
-- mlx/mlx/backend/cuda/custom_kernel.cpp
-- mlx/mlx/backend/cuda/cutlass_utils.cuh
-- mlx/mlx/backend/cuda/delayload.cpp
-- mlx/mlx/backend/cuda/device.cpp
-- mlx/mlx/backend/cuda/device.h
-- mlx/mlx/backend/cuda/device/atomic_ops.cuh
-- mlx/mlx/backend/cuda/device/binary_ops.cuh
-- mlx/mlx/backend/cuda/device/cast_op.cuh
-- mlx/mlx/backend/cuda/device/complex.cuh
-- mlx/mlx/backend/cuda/device/config.h
-- mlx/mlx/backend/cuda/device/fp16_math.cuh
-- mlx/mlx/backend/cuda/device/gather.cuh
-- mlx/mlx/backend/cuda/device/gather_axis.cuh
-- mlx/mlx/backend/cuda/device/indexing.cuh
-- mlx/mlx/backend/cuda/device/scatter.cuh
-- mlx/mlx/backend/cuda/device/scatter_axis.cuh
-- mlx/mlx/backend/cuda/device/scatter_ops.cuh
-- mlx/mlx/backend/cuda/device/ternary_ops.cuh
-- mlx/mlx/backend/cuda/device/unary_ops.cuh
-- mlx/mlx/backend/cuda/device/utils.cuh
-- mlx/mlx/backend/cuda/device_info.cpp
-- mlx/mlx/backend/cuda/distributed.cu
-- mlx/mlx/backend/cuda/eval.cpp
-- mlx/mlx/backend/cuda/event.cu
-- mlx/mlx/backend/cuda/event.h
-- mlx/mlx/backend/cuda/fence.cpp
-- mlx/mlx/backend/cuda/gemms/cublas_gemm.cpp
-- mlx/mlx/backend/cuda/gemms/cublas_gemm.h
-- mlx/mlx/backend/cuda/gemms/cublas_gemm_batched_12_0.cpp
-- mlx/mlx/backend/cuda/gemms/cublas_gemm_batched_12_9.cu
-- mlx/mlx/backend/cuda/gemms/gemv.cu
-- mlx/mlx/backend/cuda/gemms/gemv.h
-- mlx/mlx/backend/cuda/gemms/grouped_gemm.h
-- mlx/mlx/backend/cuda/gemms/grouped_gemm_unaligned.cu
-- mlx/mlx/backend/cuda/indexing.cpp
-- mlx/mlx/backend/cuda/jit_module.cpp
-- mlx/mlx/backend/cuda/jit_module.h
-- mlx/mlx/backend/cuda/kernel_utils.cu
-- mlx/mlx/backend/cuda/kernel_utils.cuh
-- mlx/mlx/backend/cuda/layer_norm.cu
-- mlx/mlx/backend/cuda/load.cpp
-- mlx/mlx/backend/cuda/logsumexp.cu
-- mlx/mlx/backend/cuda/lru_cache.h
-- mlx/mlx/backend/cuda/matmul.cpp
-- mlx/mlx/backend/cuda/no_cuda.cpp
-- mlx/mlx/backend/cuda/primitives.cpp
-- mlx/mlx/backend/cuda/quantized/affine_quantize.cu
-- mlx/mlx/backend/cuda/quantized/convert_fp8.cu
-- mlx/mlx/backend/cuda/quantized/cublas_qqmm.cpp
-- mlx/mlx/backend/cuda/quantized/cublas_qqmm.h
-- mlx/mlx/backend/cuda/quantized/cuda_fp4.h
-- mlx/mlx/backend/cuda/quantized/fp_quantize.cu
-- mlx/mlx/backend/cuda/quantized/mxfp8_quantize.cuh
-- mlx/mlx/backend/cuda/quantized/no_qqmm_impl.cpp
-- mlx/mlx/backend/cuda/quantized/nvfp4_quantize.cuh
-- mlx/mlx/backend/cuda/quantized/qmv.cu
-- mlx/mlx/backend/cuda/quantized/qmv.h
-- mlx/mlx/backend/cuda/quantized/qqmm.cpp
-- mlx/mlx/backend/cuda/quantized/qqmm_impl.cpp
-- mlx/mlx/backend/cuda/quantized/qqmm_impl.h
-- mlx/mlx/backend/cuda/quantized/qqmm_utils.cu
-- mlx/mlx/backend/cuda/quantized/qqmm_utils.h
-- mlx/mlx/backend/cuda/quantized/quantized.cpp
-- mlx/mlx/backend/cuda/quantized/quantized.h
-- mlx/mlx/backend/cuda/quantized/quantized_utils.cuh
-- mlx/mlx/backend/cuda/quantized/quantized_utils.h
-- mlx/mlx/backend/cuda/random.cu
-- mlx/mlx/backend/cuda/reduce.cu
-- mlx/mlx/backend/cuda/reduce/all_reduce.cu
-- mlx/mlx/backend/cuda/reduce/col_reduce.cu
-- mlx/mlx/backend/cuda/reduce/init_reduce.cu
-- mlx/mlx/backend/cuda/reduce/reduce.cuh
-- mlx/mlx/backend/cuda/reduce/reduce_ops.cuh
-- mlx/mlx/backend/cuda/reduce/reduce_utils.cuh
-- mlx/mlx/backend/cuda/reduce/row_reduce.cu
-- mlx/mlx/backend/cuda/rms_norm.cu
-- mlx/mlx/backend/cuda/rope.cu
-- mlx/mlx/backend/cuda/scaled_dot_product_attention.cpp
-- mlx/mlx/backend/cuda/scaled_dot_product_attention.cu
-- mlx/mlx/backend/cuda/scan.cu
-- mlx/mlx/backend/cuda/slicing.cpp
-- mlx/mlx/backend/cuda/softmax.cu
-- mlx/mlx/backend/cuda/sort.cu
-- mlx/mlx/backend/cuda/steel/defines.cuh
-- mlx/mlx/backend/cuda/steel/gemm.cuh
-- mlx/mlx/backend/cuda/steel/mma.cuh
-- mlx/mlx/backend/cuda/steel/tiles.cuh
-- mlx/mlx/backend/cuda/steel/utils.cuh
-- mlx/mlx/backend/cuda/ternary.cu
-- mlx/mlx/backend/cuda/unary/CMakeLists.txt
-- mlx/mlx/backend/cuda/unary/abs.cu
-- mlx/mlx/backend/cuda/unary/arccos.cu
-- mlx/mlx/backend/cuda/unary/arccosh.cu
-- mlx/mlx/backend/cuda/unary/arcsin.cu
-- mlx/mlx/backend/cuda/unary/arcsinh.cu
-- mlx/mlx/backend/cuda/unary/arctan.cu
-- mlx/mlx/backend/cuda/unary/arctanh.cu
-- mlx/mlx/backend/cuda/unary/bitwise_invert.cu
-- mlx/mlx/backend/cuda/unary/ceil.cu
-- mlx/mlx/backend/cuda/unary/conjugate.cu
-- mlx/mlx/backend/cuda/unary/cos.cu
-- mlx/mlx/backend/cuda/unary/cosh.cu
-- mlx/mlx/backend/cuda/unary/erf.cu
-- mlx/mlx/backend/cuda/unary/erf_inv.cu
-- mlx/mlx/backend/cuda/unary/exp.cu
-- mlx/mlx/backend/cuda/unary/expm1.cu
-- mlx/mlx/backend/cuda/unary/floor.cu
-- mlx/mlx/backend/cuda/unary/imag.cu
-- mlx/mlx/backend/cuda/unary/log.cu
-- mlx/mlx/backend/cuda/unary/log1p.cu
-- mlx/mlx/backend/cuda/unary/logical_not.cu
-- mlx/mlx/backend/cuda/unary/negative.cu
-- mlx/mlx/backend/cuda/unary/real.cu
-- mlx/mlx/backend/cuda/unary/round.cu
-- mlx/mlx/backend/cuda/unary/sigmoid.cu
-- mlx/mlx/backend/cuda/unary/sign.cu
-- mlx/mlx/backend/cuda/unary/sin.cu
-- mlx/mlx/backend/cuda/unary/sinh.cu
-- mlx/mlx/backend/cuda/unary/sqrt.cu
-- mlx/mlx/backend/cuda/unary/square.cu
-- mlx/mlx/backend/cuda/unary/tan.cu
-- mlx/mlx/backend/cuda/unary/tanh.cu
-- mlx/mlx/backend/cuda/unary/unary.cuh
-- mlx/mlx/backend/cuda/utils.cpp
-- mlx/mlx/backend/cuda/utils.h
-- mlx/mlx/backend/cuda/vector_types.cuh
-- mlx/mlx/backend/cuda/worker.cpp
-- mlx/mlx/backend/cuda/worker.h
-- mlx/mlx/backend/gpu/CMakeLists.txt
-- mlx/mlx/backend/gpu/copy.cpp
-- mlx/mlx/backend/gpu/copy.h
-- mlx/mlx/backend/gpu/device_info.h
-- mlx/mlx/backend/gpu/eval.h
-- mlx/mlx/backend/gpu/primitives.cpp
-- mlx/mlx/backend/gpu/slicing.cpp
-- mlx/mlx/backend/gpu/slicing.h
-- mlx/mlx/backend/metal/CMakeLists.txt
-- mlx/mlx/backend/metal/allocator.cpp
-- mlx/mlx/backend/metal/allocator.h
-- mlx/mlx/backend/metal/binary.cpp
-- mlx/mlx/backend/metal/binary.h
-- mlx/mlx/backend/metal/compiled.cpp
-- mlx/mlx/backend/metal/conv.cpp
-- mlx/mlx/backend/metal/copy.cpp
-- mlx/mlx/backend/metal/custom_kernel.cpp
-- mlx/mlx/backend/metal/device.cpp
-- mlx/mlx/backend/metal/device.h
-- mlx/mlx/backend/metal/device_info.cpp
-- mlx/mlx/backend/metal/distributed.cpp
-- mlx/mlx/backend/metal/eval.cpp
-- mlx/mlx/backend/metal/event.cpp
-- mlx/mlx/backend/metal/fence.cpp
-- mlx/mlx/backend/metal/fft.cpp
-- mlx/mlx/backend/metal/hadamard.cpp
-- mlx/mlx/backend/metal/indexing.cpp
-- mlx/mlx/backend/metal/jit/includes.h
-- mlx/mlx/backend/metal/jit/indexing.h
-- mlx/mlx/backend/metal/jit_kernels.cpp
-- mlx/mlx/backend/metal/kernels.h
-- mlx/mlx/backend/metal/kernels/CMakeLists.txt
-- mlx/mlx/backend/metal/kernels/arange.h
-- mlx/mlx/backend/metal/kernels/arange.metal
-- mlx/mlx/backend/metal/kernels/arg_reduce.metal
-- mlx/mlx/backend/metal/kernels/atomic.h
-- mlx/mlx/backend/metal/kernels/bf16.h
-- mlx/mlx/backend/metal/kernels/bf16_math.h
-- mlx/mlx/backend/metal/kernels/binary.h
-- mlx/mlx/backend/metal/kernels/binary.metal
-- mlx/mlx/backend/metal/kernels/binary_ops.h
-- mlx/mlx/backend/metal/kernels/binary_two.h
-- mlx/mlx/backend/metal/kernels/binary_two.metal
-- mlx/mlx/backend/metal/kernels/cexpf.h
-- mlx/mlx/backend/metal/kernels/complex.h
-- mlx/mlx/backend/metal/kernels/conv.metal
-- mlx/mlx/backend/metal/kernels/copy.h
-- mlx/mlx/backend/metal/kernels/copy.metal
-- mlx/mlx/backend/metal/kernels/defines.h
-- mlx/mlx/backend/metal/kernels/erf.h
-- mlx/mlx/backend/metal/kernels/expm1f.h
-- mlx/mlx/backend/metal/kernels/fence.metal
-- mlx/mlx/backend/metal/kernels/fft.h
-- mlx/mlx/backend/metal/kernels/fft.metal
-- mlx/mlx/backend/metal/kernels/fft/radix.h
-- mlx/mlx/backend/metal/kernels/fft/readwrite.h
-- mlx/mlx/backend/metal/kernels/fp4.h
-- mlx/mlx/backend/metal/kernels/fp8.h
-- mlx/mlx/backend/metal/kernels/fp_quantized.h
-- mlx/mlx/backend/metal/kernels/fp_quantized.metal
-- mlx/mlx/backend/metal/kernels/fp_quantized_nax.h
-- mlx/mlx/backend/metal/kernels/fp_quantized_nax.metal
-- mlx/mlx/backend/metal/kernels/gemv.metal
-- mlx/mlx/backend/metal/kernels/gemv_masked.h
-- mlx/mlx/backend/metal/kernels/gemv_masked.metal
-- mlx/mlx/backend/metal/kernels/hadamard.h
-- mlx/mlx/backend/metal/kernels/indexing/gather.h
-- mlx/mlx/backend/metal/kernels/indexing/gather_axis.h
-- mlx/mlx/backend/metal/kernels/indexing/gather_front.h
-- mlx/mlx/backend/metal/kernels/indexing/indexing.h
-- mlx/mlx/backend/metal/kernels/indexing/masked_scatter.h
-- mlx/mlx/backend/metal/kernels/indexing/scatter.h
-- mlx/mlx/backend/metal/kernels/indexing/scatter_axis.h
-- mlx/mlx/backend/metal/kernels/layer_norm.metal
-- mlx/mlx/backend/metal/kernels/logging.h
-- mlx/mlx/backend/metal/kernels/logsumexp.h
-- mlx/mlx/backend/metal/kernels/logsumexp.metal
-- mlx/mlx/backend/metal/kernels/quantized.h
-- mlx/mlx/backend/metal/kernels/quantized.metal
-- mlx/mlx/backend/metal/kernels/quantized_nax.h
-- mlx/mlx/backend/metal/kernels/quantized_nax.metal
-- mlx/mlx/backend/metal/kernels/quantized_utils.h
-- mlx/mlx/backend/metal/kernels/random.metal
-- mlx/mlx/backend/metal/kernels/reduce.h
-- mlx/mlx/backend/metal/kernels/reduce.metal
-- mlx/mlx/backend/metal/kernels/reduce_utils.h
-- mlx/mlx/backend/metal/kernels/reduction/ops.h
-- mlx/mlx/backend/metal/kernels/reduction/reduce_all.h
-- mlx/mlx/backend/metal/kernels/reduction/reduce_col.h
-- mlx/mlx/backend/metal/kernels/reduction/reduce_init.h
-- mlx/mlx/backend/metal/kernels/reduction/reduce_row.h
-- mlx/mlx/backend/metal/kernels/rms_norm.metal
-- mlx/mlx/backend/metal/kernels/rope.metal
-- mlx/mlx/backend/metal/kernels/scaled_dot_product_attention.metal
-- mlx/mlx/backend/metal/kernels/scan.h
-- mlx/mlx/backend/metal/kernels/scan.metal
-- mlx/mlx/backend/metal/kernels/sdpa_vector.h
-- mlx/mlx/backend/metal/kernels/softmax.h
-- mlx/mlx/backend/metal/kernels/softmax.metal
-- mlx/mlx/backend/metal/kernels/sort.h
-- mlx/mlx/backend/metal/kernels/sort.metal
-- mlx/mlx/backend/metal/kernels/steel/attn/attn.h
-- mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.h
-- mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.metal
-- mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.h
-- mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.metal
-- mlx/mlx/backend/metal/kernels/steel/attn/loader.h
-- mlx/mlx/backend/metal/kernels/steel/attn/mma.h
-- mlx/mlx/backend/metal/kernels/steel/attn/nax.h
-- mlx/mlx/backend/metal/kernels/steel/attn/params.h
-- mlx/mlx/backend/metal/kernels/steel/attn/transforms.h
-- mlx/mlx/backend/metal/kernels/steel/conv/conv.h
-- mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.h
-- mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.metal
-- mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.h
-- mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.metal
-- mlx/mlx/backend/metal/kernels/steel/conv/loader.h
-- mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h
-- mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h
-- mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h
-- mlx/mlx/backend/metal/kernels/steel/conv/params.h
-- mlx/mlx/backend/metal/kernels/steel/defines.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/gemm.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/gemm_nax.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk_nax.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk_nax.metal
-- mlx/mlx/backend/metal/kernels/steel/gemm/loader.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/mma.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/nax.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/params.h
-- mlx/mlx/backend/metal/kernels/steel/gemm/transforms.h
-- mlx/mlx/backend/metal/kernels/steel/utils.h
-- mlx/mlx/backend/metal/kernels/steel/utils/integral_constant.h
-- mlx/mlx/backend/metal/kernels/steel/utils/type_traits.h
-- mlx/mlx/backend/metal/kernels/ternary.h
-- mlx/mlx/backend/metal/kernels/ternary.metal
-- mlx/mlx/backend/metal/kernels/ternary_ops.h
-- mlx/mlx/backend/metal/kernels/unary.h
-- mlx/mlx/backend/metal/kernels/unary.metal
-- mlx/mlx/backend/metal/kernels/unary_ops.h
-- mlx/mlx/backend/metal/kernels/utils.h
-- mlx/mlx/backend/metal/logsumexp.cpp
-- mlx/mlx/backend/metal/make_compiled_preamble.sh
-- mlx/mlx/backend/metal/matmul.cpp
-- mlx/mlx/backend/metal/matmul.h
-- mlx/mlx/backend/metal/metal.cpp
-- mlx/mlx/backend/metal/metal.h
-- mlx/mlx/backend/metal/no_metal.cpp
-- mlx/mlx/backend/metal/nojit_kernels.cpp
-- mlx/mlx/backend/metal/normalization.cpp
-- mlx/mlx/backend/metal/primitives.cpp
-- mlx/mlx/backend/metal/quantized.cpp
-- mlx/mlx/backend/metal/reduce.cpp
-- mlx/mlx/backend/metal/reduce.h
-- mlx/mlx/backend/metal/resident.cpp
-- mlx/mlx/backend/metal/resident.h
-- mlx/mlx/backend/metal/rope.cpp
-- mlx/mlx/backend/metal/scaled_dot_product_attention.cpp
-- mlx/mlx/backend/metal/scan.cpp
-- mlx/mlx/backend/metal/scan.h
-- mlx/mlx/backend/metal/slicing.cpp
-- mlx/mlx/backend/metal/softmax.cpp
-- mlx/mlx/backend/metal/sort.cpp
-- mlx/mlx/backend/metal/ternary.cpp
-- mlx/mlx/backend/metal/ternary.h
-- mlx/mlx/backend/metal/unary.cpp
-- mlx/mlx/backend/metal/unary.h
-- mlx/mlx/backend/metal/utils.cpp
-- mlx/mlx/backend/metal/utils.h
-- mlx/mlx/backend/no_cpu/CMakeLists.txt
-- mlx/mlx/backend/no_cpu/compiled.cpp
-- mlx/mlx/backend/no_cpu/device_info.cpp
-- mlx/mlx/backend/no_cpu/primitives.cpp
-- mlx/mlx/backend/no_gpu/CMakeLists.txt
-- mlx/mlx/backend/no_gpu/allocator.cpp
-- mlx/mlx/backend/no_gpu/apple_memory.h
-- mlx/mlx/backend/no_gpu/device_info.cpp
-- mlx/mlx/backend/no_gpu/eval.cpp
-- mlx/mlx/backend/no_gpu/event.cpp
-- mlx/mlx/backend/no_gpu/fence.cpp
-- mlx/mlx/backend/no_gpu/linux_memory.h
-- mlx/mlx/backend/no_gpu/primitives.cpp
-- mlx/mlx/compile.cpp
-- mlx/mlx/compile.h
-- mlx/mlx/compile_impl.h
-- mlx/mlx/device.cpp
-- mlx/mlx/device.h
-- mlx/mlx/distributed/CMakeLists.txt
-- mlx/mlx/distributed/distributed.cpp
-- mlx/mlx/distributed/distributed.h
-- mlx/mlx/distributed/distributed_impl.h
-- mlx/mlx/distributed/jaccl/CMakeLists.txt
-- mlx/mlx/distributed/jaccl/jaccl.cpp
-- mlx/mlx/distributed/jaccl/jaccl.h
-- mlx/mlx/distributed/jaccl/mesh.cpp
-- mlx/mlx/distributed/jaccl/mesh.h
-- mlx/mlx/distributed/jaccl/no_jaccl.cpp
-- mlx/mlx/distributed/jaccl/ring.cpp
-- mlx/mlx/distributed/jaccl/ring.h
-- mlx/mlx/distributed/jaccl/utils.cpp
-- mlx/mlx/distributed/jaccl/utils.h
-- mlx/mlx/distributed/mpi/CMakeLists.txt
-- mlx/mlx/distributed/mpi/mpi.cpp
-- mlx/mlx/distributed/mpi/mpi.h
-- mlx/mlx/distributed/mpi/mpi_declarations.h
-- mlx/mlx/distributed/mpi/no_mpi.cpp
-- mlx/mlx/distributed/nccl/CMakeLists.txt
-- mlx/mlx/distributed/nccl/nccl.cpp
-- mlx/mlx/distributed/nccl/nccl.h
-- mlx/mlx/distributed/nccl/nccl_stub/CMakeLists.txt
-- mlx/mlx/distributed/nccl/nccl_stub/nccl_stubs.cpp
-- mlx/mlx/distributed/nccl/no_nccl.cpp
-- mlx/mlx/distributed/ops.cpp
-- mlx/mlx/distributed/ops.h
-- mlx/mlx/distributed/primitives.cpp
-- mlx/mlx/distributed/primitives.h
-- mlx/mlx/distributed/reduction_ops.h
-- mlx/mlx/distributed/ring/CMakeLists.txt
-- mlx/mlx/distributed/ring/no_ring.cpp
-- mlx/mlx/distributed/ring/ring.cpp
-- mlx/mlx/distributed/ring/ring.h
-- mlx/mlx/distributed/utils.cpp
-- mlx/mlx/distributed/utils.h
-- mlx/mlx/dtype.cpp
-- mlx/mlx/dtype.h
-- mlx/mlx/dtype_utils.cpp
-- mlx/mlx/dtype_utils.h
-- mlx/mlx/einsum.cpp
-- mlx/mlx/einsum.h
-- mlx/mlx/event.h
-- mlx/mlx/export.cpp
-- mlx/mlx/export.h
-- mlx/mlx/export_impl.h
-- mlx/mlx/fast.cpp
-- mlx/mlx/fast.h
-- mlx/mlx/fast_primitives.h
-- mlx/mlx/fence.h
-- mlx/mlx/fft.cpp
-- mlx/mlx/fft.h
-- mlx/mlx/graph_utils.cpp
-- mlx/mlx/graph_utils.h
-- mlx/mlx/io.h
-- mlx/mlx/io/CMakeLists.txt
-- mlx/mlx/io/gguf.cpp
-- mlx/mlx/io/gguf.h
-- mlx/mlx/io/gguf_quants.cpp
-- mlx/mlx/io/load.cpp
-- mlx/mlx/io/load.h
-- mlx/mlx/io/no_gguf.cpp
-- mlx/mlx/io/no_safetensors.cpp
-- mlx/mlx/io/safetensors.cpp
-- mlx/mlx/linalg.cpp
-- mlx/mlx/linalg.h
-- mlx/mlx/memory.h
-- mlx/mlx/mlx.h
-- mlx/mlx/ops.cpp
-- mlx/mlx/ops.h
-- mlx/mlx/primitives.cpp
-- mlx/mlx/primitives.h
-- mlx/mlx/random.cpp
-- mlx/mlx/random.h
-- mlx/mlx/scheduler.cpp
-- mlx/mlx/scheduler.h
-- mlx/mlx/small_vector.h
-- mlx/mlx/stream.h
-- mlx/mlx/threadpool.h
-- mlx/mlx/transforms.cpp
-- mlx/mlx/transforms.h
-- mlx/mlx/transforms_impl.h
-- mlx/mlx/types/bf16.h
-- mlx/mlx/types/complex.h
-- mlx/mlx/types/fp16.h
-- mlx/mlx/types/half_types.h
-- mlx/mlx/types/limits.h
-- mlx/mlx/utils.cpp
-- mlx/mlx/utils.h
-- mlx/mlx/version.cpp
-- mlx/mlx/version.h
+- submodules/mlx-onnx/CMakeLists.txt
+- submodules/mlx-onnx/LICENSE
+- submodules/mlx-onnx/include/mlx/ir.hpp
+- submodules/mlx-onnx/src/api.cpp
+- submodules/mlx-onnx/src/compat.cpp
+- submodules/mlx-onnx/src/detail.hpp
+- submodules/mlx-onnx/src/export.cpp
+- submodules/mlx-onnx/src/io.cpp
+- submodules/mlx-onnx/src/json.hpp
+- submodules/mlx-onnx/src/lowering.cpp
+- submodules/mlx-onnx/src/mappings.cpp
+- submodules/mlx-onnx/src/mappings.hpp
+- submodules/mlx-onnx/src/onnx.cpp
+- submodules/mlx-onnx/src/shared.cpp
+- submodules/mlx/CMakeLists.txt
+- submodules/mlx/cmake/FindCUDNN.cmake
+- submodules/mlx/cmake/FindNCCL.cmake
+- submodules/mlx/cmake/Findnvpl.cmake
+- submodules/mlx/cmake/extension.cmake
+- submodules/mlx/mlx.pc.in
+- submodules/mlx/mlx/3rdparty/.clang-format
+- submodules/mlx/mlx/3rdparty/pocketfft.h
+- submodules/mlx/mlx/CMakeLists.txt
+- submodules/mlx/mlx/allocator.h
+- submodules/mlx/mlx/api.h
+- submodules/mlx/mlx/array.cpp
+- submodules/mlx/mlx/array.h
+- submodules/mlx/mlx/backend/common/CMakeLists.txt
+- submodules/mlx/mlx/backend/common/binary.h
+- submodules/mlx/mlx/backend/common/broadcasting.cpp
+- submodules/mlx/mlx/backend/common/broadcasting.h
+- submodules/mlx/mlx/backend/common/buffer_cache.h
+- submodules/mlx/mlx/backend/common/common.cpp
+- submodules/mlx/mlx/backend/common/compiled.cpp
+- submodules/mlx/mlx/backend/common/compiled.h
+- submodules/mlx/mlx/backend/common/copy.h
+- submodules/mlx/mlx/backend/common/hadamard.h
+- submodules/mlx/mlx/backend/common/load.cpp
+- submodules/mlx/mlx/backend/common/matmul.h
+- submodules/mlx/mlx/backend/common/reduce.cpp
+- submodules/mlx/mlx/backend/common/reduce.h
+- submodules/mlx/mlx/backend/common/slicing.cpp
+- submodules/mlx/mlx/backend/common/slicing.h
+- submodules/mlx/mlx/backend/common/ternary.h
+- submodules/mlx/mlx/backend/common/unary.h
+- submodules/mlx/mlx/backend/common/utils.cpp
+- submodules/mlx/mlx/backend/common/utils.h
+- submodules/mlx/mlx/backend/cpu/CMakeLists.txt
+- submodules/mlx/mlx/backend/cpu/arange.h
+- submodules/mlx/mlx/backend/cpu/arg_reduce.cpp
+- submodules/mlx/mlx/backend/cpu/binary.cpp
+- submodules/mlx/mlx/backend/cpu/binary.h
+- submodules/mlx/mlx/backend/cpu/binary_ops.h
+- submodules/mlx/mlx/backend/cpu/binary_two.h
+- submodules/mlx/mlx/backend/cpu/cholesky.cpp
+- submodules/mlx/mlx/backend/cpu/compiled.cpp
+- submodules/mlx/mlx/backend/cpu/compiled_preamble.h
+- submodules/mlx/mlx/backend/cpu/conv.cpp
+- submodules/mlx/mlx/backend/cpu/copy.cpp
+- submodules/mlx/mlx/backend/cpu/copy.h
+- submodules/mlx/mlx/backend/cpu/device_info.cpp
+- submodules/mlx/mlx/backend/cpu/device_info.h
+- submodules/mlx/mlx/backend/cpu/distributed.cpp
+- submodules/mlx/mlx/backend/cpu/eig.cpp
+- submodules/mlx/mlx/backend/cpu/eigh.cpp
+- submodules/mlx/mlx/backend/cpu/encoder.cpp
+- submodules/mlx/mlx/backend/cpu/encoder.h
+- submodules/mlx/mlx/backend/cpu/eval.cpp
+- submodules/mlx/mlx/backend/cpu/eval.h
+- submodules/mlx/mlx/backend/cpu/fft.cpp
+- submodules/mlx/mlx/backend/cpu/gemm.h
+- submodules/mlx/mlx/backend/cpu/gemms/bnns.cpp
+- submodules/mlx/mlx/backend/cpu/gemms/cblas.cpp
+- submodules/mlx/mlx/backend/cpu/gemms/simd_bf16.cpp
+- submodules/mlx/mlx/backend/cpu/gemms/simd_fp16.cpp
+- submodules/mlx/mlx/backend/cpu/gemms/simd_gemm.h
+- submodules/mlx/mlx/backend/cpu/hadamard.cpp
+- submodules/mlx/mlx/backend/cpu/indexing.cpp
+- submodules/mlx/mlx/backend/cpu/inverse.cpp
+- submodules/mlx/mlx/backend/cpu/jit_compiler.cpp
+- submodules/mlx/mlx/backend/cpu/jit_compiler.h
+- submodules/mlx/mlx/backend/cpu/lapack.h
+- submodules/mlx/mlx/backend/cpu/logsumexp.cpp
+- submodules/mlx/mlx/backend/cpu/luf.cpp
+- submodules/mlx/mlx/backend/cpu/make_compiled_preamble.ps1
+- submodules/mlx/mlx/backend/cpu/make_compiled_preamble.sh
+- submodules/mlx/mlx/backend/cpu/masked_mm.cpp
+- submodules/mlx/mlx/backend/cpu/matmul.cpp
+- submodules/mlx/mlx/backend/cpu/primitives.cpp
+- submodules/mlx/mlx/backend/cpu/qrf.cpp
+- submodules/mlx/mlx/backend/cpu/quantized.cpp
+- submodules/mlx/mlx/backend/cpu/reduce.cpp
+- submodules/mlx/mlx/backend/cpu/scan.cpp
+- submodules/mlx/mlx/backend/cpu/select.cpp
+- submodules/mlx/mlx/backend/cpu/simd/accelerate_fp16_simd.h
+- submodules/mlx/mlx/backend/cpu/simd/accelerate_simd.h
+- submodules/mlx/mlx/backend/cpu/simd/base_simd.h
+- submodules/mlx/mlx/backend/cpu/simd/math.h
+- submodules/mlx/mlx/backend/cpu/simd/neon_fp16_simd.h
+- submodules/mlx/mlx/backend/cpu/simd/simd.h
+- submodules/mlx/mlx/backend/cpu/simd/type.h
+- submodules/mlx/mlx/backend/cpu/slicing.h
+- submodules/mlx/mlx/backend/cpu/softmax.cpp
+- submodules/mlx/mlx/backend/cpu/sort.cpp
+- submodules/mlx/mlx/backend/cpu/svd.cpp
+- submodules/mlx/mlx/backend/cpu/ternary.h
+- submodules/mlx/mlx/backend/cpu/threefry.cpp
+- submodules/mlx/mlx/backend/cpu/threefry.h
+- submodules/mlx/mlx/backend/cpu/unary.cpp
+- submodules/mlx/mlx/backend/cpu/unary.h
+- submodules/mlx/mlx/backend/cpu/unary_ops.h
+- submodules/mlx/mlx/backend/cuda/CMakeLists.txt
+- submodules/mlx/mlx/backend/cuda/allocator.cpp
+- submodules/mlx/mlx/backend/cuda/allocator.h
+- submodules/mlx/mlx/backend/cuda/arange.cu
+- submodules/mlx/mlx/backend/cuda/arg_reduce.cu
+- submodules/mlx/mlx/backend/cuda/bin2h.cmake
+- submodules/mlx/mlx/backend/cuda/binary/CMakeLists.txt
+- submodules/mlx/mlx/backend/cuda/binary/add.cu
+- submodules/mlx/mlx/backend/cuda/binary/arctan2.cu
+- submodules/mlx/mlx/backend/cuda/binary/binary.cuh
+- submodules/mlx/mlx/backend/cuda/binary/bitwise_binary.cu
+- submodules/mlx/mlx/backend/cuda/binary/divide.cu
+- submodules/mlx/mlx/backend/cuda/binary/equal.cu
+- submodules/mlx/mlx/backend/cuda/binary/greater.cu
+- submodules/mlx/mlx/backend/cuda/binary/greater_equal.cu
+- submodules/mlx/mlx/backend/cuda/binary/less.cu
+- submodules/mlx/mlx/backend/cuda/binary/less_equal.cu
+- submodules/mlx/mlx/backend/cuda/binary/log_add_exp.cu
+- submodules/mlx/mlx/backend/cuda/binary/logical_and.cu
+- submodules/mlx/mlx/backend/cuda/binary/logical_or.cu
+- submodules/mlx/mlx/backend/cuda/binary/maximum.cu
+- submodules/mlx/mlx/backend/cuda/binary/minimum.cu
+- submodules/mlx/mlx/backend/cuda/binary/multiply.cu
+- submodules/mlx/mlx/backend/cuda/binary/not_equal.cu
+- submodules/mlx/mlx/backend/cuda/binary/power.cu
+- submodules/mlx/mlx/backend/cuda/binary/remainder.cu
+- submodules/mlx/mlx/backend/cuda/binary/subtract.cu
+- submodules/mlx/mlx/backend/cuda/binary_two.cu
+- submodules/mlx/mlx/backend/cuda/compiled.cpp
+- submodules/mlx/mlx/backend/cuda/conv.cpp
+- submodules/mlx/mlx/backend/cuda/conv/conv.h
+- submodules/mlx/mlx/backend/cuda/conv/gemm_conv.cu
+- submodules/mlx/mlx/backend/cuda/conv/gemm_grouped_conv.cu
+- submodules/mlx/mlx/backend/cuda/copy.cu
+- submodules/mlx/mlx/backend/cuda/copy/copy.cuh
+- submodules/mlx/mlx/backend/cuda/copy/copy_contiguous.cu
+- submodules/mlx/mlx/backend/cuda/copy/copy_general.cu
+- submodules/mlx/mlx/backend/cuda/copy/copy_general_dynamic.cu
+- submodules/mlx/mlx/backend/cuda/copy/copy_general_input.cu
+- submodules/mlx/mlx/backend/cuda/cublas_utils.cpp
+- submodules/mlx/mlx/backend/cuda/cublas_utils.h
+- submodules/mlx/mlx/backend/cuda/cuda.h
+- submodules/mlx/mlx/backend/cuda/cuda_utils.h
+- submodules/mlx/mlx/backend/cuda/cudnn_utils.cpp
+- submodules/mlx/mlx/backend/cuda/cudnn_utils.h
+- submodules/mlx/mlx/backend/cuda/custom_kernel.cpp
+- submodules/mlx/mlx/backend/cuda/cutlass_utils.cuh
+- submodules/mlx/mlx/backend/cuda/delayload.cpp
+- submodules/mlx/mlx/backend/cuda/device.cpp
+- submodules/mlx/mlx/backend/cuda/device.h
+- submodules/mlx/mlx/backend/cuda/device/atomic_ops.cuh
+- submodules/mlx/mlx/backend/cuda/device/binary_ops.cuh
+- submodules/mlx/mlx/backend/cuda/device/cast_op.cuh
+- submodules/mlx/mlx/backend/cuda/device/complex.cuh
+- submodules/mlx/mlx/backend/cuda/device/config.h
+- submodules/mlx/mlx/backend/cuda/device/fp16_math.cuh
+- submodules/mlx/mlx/backend/cuda/device/gather.cuh
+- submodules/mlx/mlx/backend/cuda/device/gather_axis.cuh
+- submodules/mlx/mlx/backend/cuda/device/indexing.cuh
+- submodules/mlx/mlx/backend/cuda/device/scatter.cuh
+- submodules/mlx/mlx/backend/cuda/device/scatter_axis.cuh
+- submodules/mlx/mlx/backend/cuda/device/scatter_ops.cuh
+- submodules/mlx/mlx/backend/cuda/device/ternary_ops.cuh
+- submodules/mlx/mlx/backend/cuda/device/unary_ops.cuh
+- submodules/mlx/mlx/backend/cuda/device/utils.cuh
+- submodules/mlx/mlx/backend/cuda/device_info.cpp
+- submodules/mlx/mlx/backend/cuda/distributed.cu
+- submodules/mlx/mlx/backend/cuda/eval.cpp
+- submodules/mlx/mlx/backend/cuda/event.cu
+- submodules/mlx/mlx/backend/cuda/event.h
+- submodules/mlx/mlx/backend/cuda/fence.cpp
+- submodules/mlx/mlx/backend/cuda/gemms/cublas_gemm.cpp
+- submodules/mlx/mlx/backend/cuda/gemms/cublas_gemm.h
+- submodules/mlx/mlx/backend/cuda/gemms/cublas_gemm_batched_12_0.cpp
+- submodules/mlx/mlx/backend/cuda/gemms/cublas_gemm_batched_12_9.cu
+- submodules/mlx/mlx/backend/cuda/gemms/gemv.cu
+- submodules/mlx/mlx/backend/cuda/gemms/gemv.h
+- submodules/mlx/mlx/backend/cuda/gemms/grouped_gemm.h
+- submodules/mlx/mlx/backend/cuda/gemms/grouped_gemm_unaligned.cu
+- submodules/mlx/mlx/backend/cuda/indexing.cpp
+- submodules/mlx/mlx/backend/cuda/jit_module.cpp
+- submodules/mlx/mlx/backend/cuda/jit_module.h
+- submodules/mlx/mlx/backend/cuda/kernel_utils.cu
+- submodules/mlx/mlx/backend/cuda/kernel_utils.cuh
+- submodules/mlx/mlx/backend/cuda/layer_norm.cu
+- submodules/mlx/mlx/backend/cuda/load.cpp
+- submodules/mlx/mlx/backend/cuda/logsumexp.cu
+- submodules/mlx/mlx/backend/cuda/lru_cache.h
+- submodules/mlx/mlx/backend/cuda/matmul.cpp
+- submodules/mlx/mlx/backend/cuda/no_cuda.cpp
+- submodules/mlx/mlx/backend/cuda/primitives.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/affine_quantize.cu
+- submodules/mlx/mlx/backend/cuda/quantized/convert_fp8.cu
+- submodules/mlx/mlx/backend/cuda/quantized/cublas_qqmm.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/cublas_qqmm.h
+- submodules/mlx/mlx/backend/cuda/quantized/cuda_fp4.h
+- submodules/mlx/mlx/backend/cuda/quantized/fp_quantize.cu
+- submodules/mlx/mlx/backend/cuda/quantized/mxfp8_quantize.cuh
+- submodules/mlx/mlx/backend/cuda/quantized/no_qqmm_impl.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/nvfp4_quantize.cuh
+- submodules/mlx/mlx/backend/cuda/quantized/qmv.cu
+- submodules/mlx/mlx/backend/cuda/quantized/qmv.h
+- submodules/mlx/mlx/backend/cuda/quantized/qqmm.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/qqmm_impl.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/qqmm_impl.h
+- submodules/mlx/mlx/backend/cuda/quantized/qqmm_utils.cu
+- submodules/mlx/mlx/backend/cuda/quantized/qqmm_utils.h
+- submodules/mlx/mlx/backend/cuda/quantized/quantized.cpp
+- submodules/mlx/mlx/backend/cuda/quantized/quantized.h
+- submodules/mlx/mlx/backend/cuda/quantized/quantized_utils.cuh
+- submodules/mlx/mlx/backend/cuda/quantized/quantized_utils.h
+- submodules/mlx/mlx/backend/cuda/random.cu
+- submodules/mlx/mlx/backend/cuda/reduce.cu
+- submodules/mlx/mlx/backend/cuda/reduce/all_reduce.cu
+- submodules/mlx/mlx/backend/cuda/reduce/col_reduce.cu
+- submodules/mlx/mlx/backend/cuda/reduce/init_reduce.cu
+- submodules/mlx/mlx/backend/cuda/reduce/reduce.cuh
+- submodules/mlx/mlx/backend/cuda/reduce/reduce_ops.cuh
+- submodules/mlx/mlx/backend/cuda/reduce/reduce_utils.cuh
+- submodules/mlx/mlx/backend/cuda/reduce/row_reduce.cu
+- submodules/mlx/mlx/backend/cuda/rms_norm.cu
+- submodules/mlx/mlx/backend/cuda/rope.cu
+- submodules/mlx/mlx/backend/cuda/scaled_dot_product_attention.cpp
+- submodules/mlx/mlx/backend/cuda/scaled_dot_product_attention.cu
+- submodules/mlx/mlx/backend/cuda/scan.cu
+- submodules/mlx/mlx/backend/cuda/slicing.cpp
+- submodules/mlx/mlx/backend/cuda/softmax.cu
+- submodules/mlx/mlx/backend/cuda/sort.cu
+- submodules/mlx/mlx/backend/cuda/steel/defines.cuh
+- submodules/mlx/mlx/backend/cuda/steel/gemm.cuh
+- submodules/mlx/mlx/backend/cuda/steel/mma.cuh
+- submodules/mlx/mlx/backend/cuda/steel/tiles.cuh
+- submodules/mlx/mlx/backend/cuda/steel/utils.cuh
+- submodules/mlx/mlx/backend/cuda/ternary.cu
+- submodules/mlx/mlx/backend/cuda/unary/CMakeLists.txt
+- submodules/mlx/mlx/backend/cuda/unary/abs.cu
+- submodules/mlx/mlx/backend/cuda/unary/arccos.cu
+- submodules/mlx/mlx/backend/cuda/unary/arccosh.cu
+- submodules/mlx/mlx/backend/cuda/unary/arcsin.cu
+- submodules/mlx/mlx/backend/cuda/unary/arcsinh.cu
+- submodules/mlx/mlx/backend/cuda/unary/arctan.cu
+- submodules/mlx/mlx/backend/cuda/unary/arctanh.cu
+- submodules/mlx/mlx/backend/cuda/unary/bitwise_invert.cu
+- submodules/mlx/mlx/backend/cuda/unary/ceil.cu
+- submodules/mlx/mlx/backend/cuda/unary/conjugate.cu
+- submodules/mlx/mlx/backend/cuda/unary/cos.cu
+- submodules/mlx/mlx/backend/cuda/unary/cosh.cu
+- submodules/mlx/mlx/backend/cuda/unary/erf.cu
+- submodules/mlx/mlx/backend/cuda/unary/erf_inv.cu
+- submodules/mlx/mlx/backend/cuda/unary/exp.cu
+- submodules/mlx/mlx/backend/cuda/unary/expm1.cu
+- submodules/mlx/mlx/backend/cuda/unary/floor.cu
+- submodules/mlx/mlx/backend/cuda/unary/imag.cu
+- submodules/mlx/mlx/backend/cuda/unary/log.cu
+- submodules/mlx/mlx/backend/cuda/unary/log1p.cu
+- submodules/mlx/mlx/backend/cuda/unary/logical_not.cu
+- submodules/mlx/mlx/backend/cuda/unary/negative.cu
+- submodules/mlx/mlx/backend/cuda/unary/real.cu
+- submodules/mlx/mlx/backend/cuda/unary/round.cu
+- submodules/mlx/mlx/backend/cuda/unary/sigmoid.cu
+- submodules/mlx/mlx/backend/cuda/unary/sign.cu
+- submodules/mlx/mlx/backend/cuda/unary/sin.cu
+- submodules/mlx/mlx/backend/cuda/unary/sinh.cu
+- submodules/mlx/mlx/backend/cuda/unary/sqrt.cu
+- submodules/mlx/mlx/backend/cuda/unary/square.cu
+- submodules/mlx/mlx/backend/cuda/unary/tan.cu
+- submodules/mlx/mlx/backend/cuda/unary/tanh.cu
+- submodules/mlx/mlx/backend/cuda/unary/unary.cuh
+- submodules/mlx/mlx/backend/cuda/utils.cpp
+- submodules/mlx/mlx/backend/cuda/utils.h
+- submodules/mlx/mlx/backend/cuda/vector_types.cuh
+- submodules/mlx/mlx/backend/cuda/worker.cpp
+- submodules/mlx/mlx/backend/cuda/worker.h
+- submodules/mlx/mlx/backend/gpu/CMakeLists.txt
+- submodules/mlx/mlx/backend/gpu/copy.cpp
+- submodules/mlx/mlx/backend/gpu/copy.h
+- submodules/mlx/mlx/backend/gpu/device_info.h
+- submodules/mlx/mlx/backend/gpu/eval.h
+- submodules/mlx/mlx/backend/gpu/primitives.cpp
+- submodules/mlx/mlx/backend/gpu/slicing.cpp
+- submodules/mlx/mlx/backend/gpu/slicing.h
+- submodules/mlx/mlx/backend/metal/CMakeLists.txt
+- submodules/mlx/mlx/backend/metal/allocator.cpp
+- submodules/mlx/mlx/backend/metal/allocator.h
+- submodules/mlx/mlx/backend/metal/binary.cpp
+- submodules/mlx/mlx/backend/metal/binary.h
+- submodules/mlx/mlx/backend/metal/compiled.cpp
+- submodules/mlx/mlx/backend/metal/conv.cpp
+- submodules/mlx/mlx/backend/metal/copy.cpp
+- submodules/mlx/mlx/backend/metal/custom_kernel.cpp
+- submodules/mlx/mlx/backend/metal/device.cpp
+- submodules/mlx/mlx/backend/metal/device.h
+- submodules/mlx/mlx/backend/metal/device_info.cpp
+- submodules/mlx/mlx/backend/metal/distributed.cpp
+- submodules/mlx/mlx/backend/metal/eval.cpp
+- submodules/mlx/mlx/backend/metal/event.cpp
+- submodules/mlx/mlx/backend/metal/fence.cpp
+- submodules/mlx/mlx/backend/metal/fft.cpp
+- submodules/mlx/mlx/backend/metal/hadamard.cpp
+- submodules/mlx/mlx/backend/metal/indexing.cpp
+- submodules/mlx/mlx/backend/metal/jit/includes.h
+- submodules/mlx/mlx/backend/metal/jit/indexing.h
+- submodules/mlx/mlx/backend/metal/jit_kernels.cpp
+- submodules/mlx/mlx/backend/metal/kernels.h
+- submodules/mlx/mlx/backend/metal/kernels/CMakeLists.txt
+- submodules/mlx/mlx/backend/metal/kernels/arange.h
+- submodules/mlx/mlx/backend/metal/kernels/arange.metal
+- submodules/mlx/mlx/backend/metal/kernels/arg_reduce.metal
+- submodules/mlx/mlx/backend/metal/kernels/atomic.h
+- submodules/mlx/mlx/backend/metal/kernels/bf16.h
+- submodules/mlx/mlx/backend/metal/kernels/bf16_math.h
+- submodules/mlx/mlx/backend/metal/kernels/binary.h
+- submodules/mlx/mlx/backend/metal/kernels/binary.metal
+- submodules/mlx/mlx/backend/metal/kernels/binary_ops.h
+- submodules/mlx/mlx/backend/metal/kernels/binary_two.h
+- submodules/mlx/mlx/backend/metal/kernels/binary_two.metal
+- submodules/mlx/mlx/backend/metal/kernels/cexpf.h
+- submodules/mlx/mlx/backend/metal/kernels/complex.h
+- submodules/mlx/mlx/backend/metal/kernels/conv.metal
+- submodules/mlx/mlx/backend/metal/kernels/copy.h
+- submodules/mlx/mlx/backend/metal/kernels/copy.metal
+- submodules/mlx/mlx/backend/metal/kernels/defines.h
+- submodules/mlx/mlx/backend/metal/kernels/erf.h
+- submodules/mlx/mlx/backend/metal/kernels/expm1f.h
+- submodules/mlx/mlx/backend/metal/kernels/fence.metal
+- submodules/mlx/mlx/backend/metal/kernels/fft.h
+- submodules/mlx/mlx/backend/metal/kernels/fft.metal
+- submodules/mlx/mlx/backend/metal/kernels/fft/radix.h
+- submodules/mlx/mlx/backend/metal/kernels/fft/readwrite.h
+- submodules/mlx/mlx/backend/metal/kernels/fp4.h
+- submodules/mlx/mlx/backend/metal/kernels/fp8.h
+- submodules/mlx/mlx/backend/metal/kernels/fp_quantized.h
+- submodules/mlx/mlx/backend/metal/kernels/fp_quantized.metal
+- submodules/mlx/mlx/backend/metal/kernels/fp_quantized_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/fp_quantized_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/gemv.metal
+- submodules/mlx/mlx/backend/metal/kernels/gemv_masked.h
+- submodules/mlx/mlx/backend/metal/kernels/gemv_masked.metal
+- submodules/mlx/mlx/backend/metal/kernels/hadamard.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/gather.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/gather_axis.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/gather_front.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/indexing.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/masked_scatter.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/scatter.h
+- submodules/mlx/mlx/backend/metal/kernels/indexing/scatter_axis.h
+- submodules/mlx/mlx/backend/metal/kernels/layer_norm.metal
+- submodules/mlx/mlx/backend/metal/kernels/logging.h
+- submodules/mlx/mlx/backend/metal/kernels/logsumexp.h
+- submodules/mlx/mlx/backend/metal/kernels/logsumexp.metal
+- submodules/mlx/mlx/backend/metal/kernels/quantized.h
+- submodules/mlx/mlx/backend/metal/kernels/quantized.metal
+- submodules/mlx/mlx/backend/metal/kernels/quantized_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/quantized_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/quantized_utils.h
+- submodules/mlx/mlx/backend/metal/kernels/random.metal
+- submodules/mlx/mlx/backend/metal/kernels/reduce.h
+- submodules/mlx/mlx/backend/metal/kernels/reduce.metal
+- submodules/mlx/mlx/backend/metal/kernels/reduce_utils.h
+- submodules/mlx/mlx/backend/metal/kernels/reduction/ops.h
+- submodules/mlx/mlx/backend/metal/kernels/reduction/reduce_all.h
+- submodules/mlx/mlx/backend/metal/kernels/reduction/reduce_col.h
+- submodules/mlx/mlx/backend/metal/kernels/reduction/reduce_init.h
+- submodules/mlx/mlx/backend/metal/kernels/reduction/reduce_row.h
+- submodules/mlx/mlx/backend/metal/kernels/rms_norm.metal
+- submodules/mlx/mlx/backend/metal/kernels/rope.metal
+- submodules/mlx/mlx/backend/metal/kernels/scaled_dot_product_attention.metal
+- submodules/mlx/mlx/backend/metal/kernels/scan.h
+- submodules/mlx/mlx/backend/metal/kernels/scan.metal
+- submodules/mlx/mlx/backend/metal/kernels/sdpa_vector.h
+- submodules/mlx/mlx/backend/metal/kernels/softmax.h
+- submodules/mlx/mlx/backend/metal/kernels/softmax.metal
+- submodules/mlx/mlx/backend/metal/kernels/sort.h
+- submodules/mlx/mlx/backend/metal/kernels/sort.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/attn.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/loader.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/mma.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/params.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/attn/transforms.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/conv.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/loader.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/conv/params.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/defines.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/gemm.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/gemm_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk_nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk_nax.metal
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/loader.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/mma.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/nax.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/params.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/gemm/transforms.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/utils.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/utils/integral_constant.h
+- submodules/mlx/mlx/backend/metal/kernels/steel/utils/type_traits.h
+- submodules/mlx/mlx/backend/metal/kernels/ternary.h
+- submodules/mlx/mlx/backend/metal/kernels/ternary.metal
+- submodules/mlx/mlx/backend/metal/kernels/ternary_ops.h
+- submodules/mlx/mlx/backend/metal/kernels/unary.h
+- submodules/mlx/mlx/backend/metal/kernels/unary.metal
+- submodules/mlx/mlx/backend/metal/kernels/unary_ops.h
+- submodules/mlx/mlx/backend/metal/kernels/utils.h
+- submodules/mlx/mlx/backend/metal/logsumexp.cpp
+- submodules/mlx/mlx/backend/metal/make_compiled_preamble.sh
+- submodules/mlx/mlx/backend/metal/matmul.cpp
+- submodules/mlx/mlx/backend/metal/matmul.h
+- submodules/mlx/mlx/backend/metal/metal.cpp
+- submodules/mlx/mlx/backend/metal/metal.h
+- submodules/mlx/mlx/backend/metal/no_metal.cpp
+- submodules/mlx/mlx/backend/metal/nojit_kernels.cpp
+- submodules/mlx/mlx/backend/metal/normalization.cpp
+- submodules/mlx/mlx/backend/metal/primitives.cpp
+- submodules/mlx/mlx/backend/metal/quantized.cpp
+- submodules/mlx/mlx/backend/metal/reduce.cpp
+- submodules/mlx/mlx/backend/metal/reduce.h
+- submodules/mlx/mlx/backend/metal/resident.cpp
+- submodules/mlx/mlx/backend/metal/resident.h
+- submodules/mlx/mlx/backend/metal/rope.cpp
+- submodules/mlx/mlx/backend/metal/scaled_dot_product_attention.cpp
+- submodules/mlx/mlx/backend/metal/scan.cpp
+- submodules/mlx/mlx/backend/metal/scan.h
+- submodules/mlx/mlx/backend/metal/slicing.cpp
+- submodules/mlx/mlx/backend/metal/softmax.cpp
+- submodules/mlx/mlx/backend/metal/sort.cpp
+- submodules/mlx/mlx/backend/metal/ternary.cpp
+- submodules/mlx/mlx/backend/metal/ternary.h
+- submodules/mlx/mlx/backend/metal/unary.cpp
+- submodules/mlx/mlx/backend/metal/unary.h
+- submodules/mlx/mlx/backend/metal/utils.cpp
+- submodules/mlx/mlx/backend/metal/utils.h
+- submodules/mlx/mlx/backend/no_cpu/CMakeLists.txt
+- submodules/mlx/mlx/backend/no_cpu/compiled.cpp
+- submodules/mlx/mlx/backend/no_cpu/device_info.cpp
+- submodules/mlx/mlx/backend/no_cpu/primitives.cpp
+- submodules/mlx/mlx/backend/no_gpu/CMakeLists.txt
+- submodules/mlx/mlx/backend/no_gpu/allocator.cpp
+- submodules/mlx/mlx/backend/no_gpu/apple_memory.h
+- submodules/mlx/mlx/backend/no_gpu/device_info.cpp
+- submodules/mlx/mlx/backend/no_gpu/eval.cpp
+- submodules/mlx/mlx/backend/no_gpu/event.cpp
+- submodules/mlx/mlx/backend/no_gpu/fence.cpp
+- submodules/mlx/mlx/backend/no_gpu/linux_memory.h
+- submodules/mlx/mlx/backend/no_gpu/primitives.cpp
+- submodules/mlx/mlx/compile.cpp
+- submodules/mlx/mlx/compile.h
+- submodules/mlx/mlx/compile_impl.h
+- submodules/mlx/mlx/device.cpp
+- submodules/mlx/mlx/device.h
+- submodules/mlx/mlx/distributed/CMakeLists.txt
+- submodules/mlx/mlx/distributed/distributed.cpp
+- submodules/mlx/mlx/distributed/distributed.h
+- submodules/mlx/mlx/distributed/distributed_impl.h
+- submodules/mlx/mlx/distributed/jaccl/CMakeLists.txt
+- submodules/mlx/mlx/distributed/jaccl/jaccl.cpp
+- submodules/mlx/mlx/distributed/jaccl/jaccl.h
+- submodules/mlx/mlx/distributed/jaccl/mesh.cpp
+- submodules/mlx/mlx/distributed/jaccl/mesh.h
+- submodules/mlx/mlx/distributed/jaccl/no_jaccl.cpp
+- submodules/mlx/mlx/distributed/jaccl/ring.cpp
+- submodules/mlx/mlx/distributed/jaccl/ring.h
+- submodules/mlx/mlx/distributed/jaccl/utils.cpp
+- submodules/mlx/mlx/distributed/jaccl/utils.h
+- submodules/mlx/mlx/distributed/mpi/CMakeLists.txt
+- submodules/mlx/mlx/distributed/mpi/mpi.cpp
+- submodules/mlx/mlx/distributed/mpi/mpi.h
+- submodules/mlx/mlx/distributed/mpi/mpi_declarations.h
+- submodules/mlx/mlx/distributed/mpi/no_mpi.cpp
+- submodules/mlx/mlx/distributed/nccl/CMakeLists.txt
+- submodules/mlx/mlx/distributed/nccl/nccl.cpp
+- submodules/mlx/mlx/distributed/nccl/nccl.h
+- submodules/mlx/mlx/distributed/nccl/nccl_stub/CMakeLists.txt
+- submodules/mlx/mlx/distributed/nccl/nccl_stub/nccl_stubs.cpp
+- submodules/mlx/mlx/distributed/nccl/no_nccl.cpp
+- submodules/mlx/mlx/distributed/ops.cpp
+- submodules/mlx/mlx/distributed/ops.h
+- submodules/mlx/mlx/distributed/primitives.cpp
+- submodules/mlx/mlx/distributed/primitives.h
+- submodules/mlx/mlx/distributed/reduction_ops.h
+- submodules/mlx/mlx/distributed/ring/CMakeLists.txt
+- submodules/mlx/mlx/distributed/ring/no_ring.cpp
+- submodules/mlx/mlx/distributed/ring/ring.cpp
+- submodules/mlx/mlx/distributed/ring/ring.h
+- submodules/mlx/mlx/distributed/utils.cpp
+- submodules/mlx/mlx/distributed/utils.h
+- submodules/mlx/mlx/dtype.cpp
+- submodules/mlx/mlx/dtype.h
+- submodules/mlx/mlx/dtype_utils.cpp
+- submodules/mlx/mlx/dtype_utils.h
+- submodules/mlx/mlx/einsum.cpp
+- submodules/mlx/mlx/einsum.h
+- submodules/mlx/mlx/event.h
+- submodules/mlx/mlx/export.cpp
+- submodules/mlx/mlx/export.h
+- submodules/mlx/mlx/export_impl.h
+- submodules/mlx/mlx/fast.cpp
+- submodules/mlx/mlx/fast.h
+- submodules/mlx/mlx/fast_primitives.h
+- submodules/mlx/mlx/fence.h
+- submodules/mlx/mlx/fft.cpp
+- submodules/mlx/mlx/fft.h
+- submodules/mlx/mlx/graph_utils.cpp
+- submodules/mlx/mlx/graph_utils.h
+- submodules/mlx/mlx/io.h
+- submodules/mlx/mlx/io/CMakeLists.txt
+- submodules/mlx/mlx/io/gguf.cpp
+- submodules/mlx/mlx/io/gguf.h
+- submodules/mlx/mlx/io/gguf_quants.cpp
+- submodules/mlx/mlx/io/load.cpp
+- submodules/mlx/mlx/io/load.h
+- submodules/mlx/mlx/io/no_gguf.cpp
+- submodules/mlx/mlx/io/no_safetensors.cpp
+- submodules/mlx/mlx/io/safetensors.cpp
+- submodules/mlx/mlx/linalg.cpp
+- submodules/mlx/mlx/linalg.h
+- submodules/mlx/mlx/memory.h
+- submodules/mlx/mlx/mlx.h
+- submodules/mlx/mlx/ops.cpp
+- submodules/mlx/mlx/ops.h
+- submodules/mlx/mlx/primitives.cpp
+- submodules/mlx/mlx/primitives.h
+- submodules/mlx/mlx/random.cpp
+- submodules/mlx/mlx/random.h
+- submodules/mlx/mlx/scheduler.cpp
+- submodules/mlx/mlx/scheduler.h
+- submodules/mlx/mlx/small_vector.h
+- submodules/mlx/mlx/stream.h
+- submodules/mlx/mlx/threadpool.h
+- submodules/mlx/mlx/transforms.cpp
+- submodules/mlx/mlx/transforms.h
+- submodules/mlx/mlx/transforms_impl.h
+- submodules/mlx/mlx/types/bf16.h
+- submodules/mlx/mlx/types/complex.h
+- submodules/mlx/mlx/types/fp16.h
+- submodules/mlx/mlx/types/half_types.h
+- submodules/mlx/mlx/types/limits.h
+- submodules/mlx/mlx/utils.cpp
+- submodules/mlx/mlx/utils.h
+- submodules/mlx/mlx/version.cpp
+- submodules/mlx/mlx/version.h
 homepage: https://github.com/skryl/mlx-ruby
 licenses:
 - MIT
@@ -640,14 +738,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.1'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 4.0.3
 specification_version: 4
 summary: Ruby bindings for the native MLX library
 test_files: []