├── .clang-format ├── .github ├── ISSUE_TEMPLATE │ └── bug_report.md ├── actions │ ├── build-cuda-release │ │ └── action.yml │ ├── build-docs │ │ └── action.yml │ ├── build-linux-release │ │ └── action.yml │ ├── build-linux │ │ └── action.yml │ ├── build-macos-release │ │ └── action.yml │ ├── build-macos │ │ └── action.yml │ ├── setup-linux │ │ └── action.yml │ ├── setup-macos │ │ └── action.yml │ └── test-linux │ │ └── action.yml ├── dependabot.yml ├── pull_request_template.md ├── scripts │ └── setup+build-cpp-linux-fedora-container.sh └── workflows │ ├── build_and_test.yml │ ├── documentation.yml │ ├── nightly.yml │ └── release.yml ├── .gitignore ├── .pre-commit-config.yaml ├── ACKNOWLEDGMENTS.md ├── CITATION.cff ├── CMakeLists.txt ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── benchmarks ├── cpp │ ├── CMakeLists.txt │ ├── autograd.cpp │ ├── compare_devices.cpp │ ├── irregular_strides.cpp │ ├── single_ops.cpp │ └── time_utils.h ├── numpy │ ├── single_ops.py │ └── time_utils.py └── python │ ├── batch_matmul_bench.py │ ├── blas │ ├── bench_gemm.py │ └── bench_gemv.py │ ├── comparative │ ├── README.md │ ├── bench_mlx.py │ ├── bench_torch.py │ └── compare.py │ ├── compile_bench.py │ ├── conv1d_bench.py │ ├── conv2d_bench_cpu.py │ ├── conv2d_train_bench_cpu.py │ ├── conv2d_transpose_bench_cpu.py │ ├── conv3d_bench_cpu.py │ ├── conv3d_train_bench_cpu.py │ ├── conv3d_transpose_bench_cpu.py │ ├── conv_bench.py │ ├── conv_transpose_bench.py │ ├── conv_unaligned_bench.py │ ├── distributed_bench.py │ ├── einsum_bench.py │ ├── fft_bench.py │ ├── gather_bench.py │ ├── gather_mm_bench.py │ ├── gather_qmm_bench.py │ ├── hadamard_bench.py │ ├── layer_norm_bench.py │ ├── masked_scatter.py │ ├── rms_norm_bench.py │ ├── rope_bench.py │ ├── scatter_bench.py │ ├── sdpa_bench.py │ ├── sdpa_vector_bench.py │ ├── single_ops.py │ ├── synchronize_bench.py │ └── time_utils.py ├── cmake ├── FindNCCL.cmake ├── Findnvpl.cmake └── extension.cmake ├── docs ├── .clang-format ├── .gitignore ├── .nojekyll ├── Doxyfile ├── Makefile ├── README.md ├── index.html ├── requirements.txt └── src │ ├── _static │ ├── metal_debugger │ │ ├── capture.png │ │ └── schema.png │ ├── mlx_logo.png │ └── mlx_logo_dark.png │ ├── _templates │ ├── module-base-class.rst │ ├── nn-module-template.rst │ └── optimizers-template.rst │ ├── conf.py │ ├── cpp │ └── ops.rst │ ├── dev │ ├── custom_metal_kernels.rst │ ├── extensions.rst │ ├── metal_debugger.rst │ └── mlx_in_cpp.rst │ ├── examples │ ├── linear_regression.rst │ ├── llama-inference.rst │ └── mlp.rst │ ├── index.rst │ ├── install.rst │ ├── python │ ├── array.rst │ ├── cuda.rst │ ├── data_types.rst │ ├── devices_and_streams.rst │ ├── distributed.rst │ ├── export.rst │ ├── fast.rst │ ├── fft.rst │ ├── linalg.rst │ ├── memory_management.rst │ ├── metal.rst │ ├── nn.rst │ ├── nn │ │ ├── functions.rst │ │ ├── init.rst │ │ ├── layers.rst │ │ ├── losses.rst │ │ └── module.rst │ ├── ops.rst │ ├── optimizers.rst │ ├── optimizers │ │ ├── common_optimizers.rst │ │ ├── optimizer.rst │ │ └── schedulers.rst │ ├── random.rst │ ├── transforms.rst │ └── tree_utils.rst │ └── usage │ ├── compile.rst │ ├── distributed.rst │ ├── export.rst │ ├── function_transforms.rst │ ├── indexing.rst │ ├── launching_distributed.rst │ ├── lazy_evaluation.rst │ ├── numpy.rst │ ├── quick_start.rst │ ├── saving_and_loading.rst │ ├── unified_memory.rst │ └── using_streams.rst ├── examples ├── cmake_project │ ├── CMakeLists.txt │ ├── README.md │ └── example.cpp ├── cpp │ ├── CMakeLists.txt │ ├── distributed.cpp │ ├── linear_regression.cpp │ ├── logistic_regression.cpp │ ├── metal_capture.cpp │ ├── timer.h │ └── tutorial.cpp ├── export │ ├── CMakeLists.txt │ ├── README.md │ ├── eval_mlp.cpp │ ├── eval_mlp.py │ ├── train_mlp.cpp │ └── train_mlp.py ├── extensions │ ├── CMakeLists.txt │ ├── README.md │ ├── axpby │ │ ├── axpby.cpp │ │ ├── axpby.h │ │ └── axpby.metal │ ├── bindings.cpp │ ├── mlx_sample_extensions │ │ └── __init__.py │ ├── pyproject.toml │ ├── requirements.txt │ ├── setup.py │ └── test.py └── python │ ├── linear_regression.py │ └── logistic_regression.py ├── mlx.pc.in ├── mlx ├── 3rdparty │ ├── .clang-format │ └── pocketfft.h ├── CMakeLists.txt ├── allocator.h ├── array.cpp ├── array.h ├── backend │ ├── common │ │ ├── CMakeLists.txt │ │ ├── binary.h │ │ ├── broadcasting.cpp │ │ ├── broadcasting.h │ │ ├── buffer_cache.h │ │ ├── common.cpp │ │ ├── compiled.cpp │ │ ├── compiled.h │ │ ├── copy.h │ │ ├── hadamard.h │ │ ├── load.cpp │ │ ├── matmul.h │ │ ├── reduce.cpp │ │ ├── reduce.h │ │ ├── slicing.cpp │ │ ├── slicing.h │ │ ├── ternary.h │ │ ├── unary.h │ │ ├── utils.cpp │ │ └── utils.h │ ├── cpu │ │ ├── CMakeLists.txt │ │ ├── arange.h │ │ ├── arg_reduce.cpp │ │ ├── available.cpp │ │ ├── available.h │ │ ├── binary.cpp │ │ ├── binary.h │ │ ├── binary_ops.h │ │ ├── binary_two.h │ │ ├── cholesky.cpp │ │ ├── compiled.cpp │ │ ├── compiled_preamble.h │ │ ├── conv.cpp │ │ ├── copy.cpp │ │ ├── copy.h │ │ ├── distributed.cpp │ │ ├── eig.cpp │ │ ├── eigh.cpp │ │ ├── encoder.cpp │ │ ├── encoder.h │ │ ├── eval.cpp │ │ ├── eval.h │ │ ├── fft.cpp │ │ ├── gemm.h │ │ ├── gemms │ │ │ ├── bnns.cpp │ │ │ ├── cblas.cpp │ │ │ ├── simd_bf16.cpp │ │ │ ├── simd_fp16.cpp │ │ │ └── simd_gemm.h │ │ ├── hadamard.cpp │ │ ├── indexing.cpp │ │ ├── inverse.cpp │ │ ├── jit_compiler.cpp │ │ ├── jit_compiler.h │ │ ├── lapack.h │ │ ├── logsumexp.cpp │ │ ├── luf.cpp │ │ ├── make_compiled_preamble.ps1 │ │ ├── make_compiled_preamble.sh │ │ ├── masked_mm.cpp │ │ ├── matmul.cpp │ │ ├── primitives.cpp │ │ ├── qrf.cpp │ │ ├── quantized.cpp │ │ ├── reduce.cpp │ │ ├── scan.cpp │ │ ├── select.cpp │ │ ├── simd │ │ │ ├── accelerate_fp16_simd.h │ │ │ ├── accelerate_simd.h │ │ │ ├── base_simd.h │ │ │ ├── math.h │ │ │ ├── neon_fp16_simd.h │ │ │ ├── simd.h │ │ │ └── type.h │ │ ├── slicing.h │ │ ├── softmax.cpp │ │ ├── sort.cpp │ │ ├── svd.cpp │ │ ├── ternary.h │ │ ├── threefry.cpp │ │ ├── threefry.h │ │ ├── unary.cpp │ │ ├── unary.h │ │ └── unary_ops.h │ ├── cuda │ │ ├── CMakeLists.txt │ │ ├── allocator.cpp │ │ ├── allocator.h │ │ ├── arange.cu │ │ ├── arg_reduce.cu │ │ ├── bin2h.cmake │ │ ├── binary │ │ │ ├── CMakeLists.txt │ │ │ ├── add.cu │ │ │ ├── arctan2.cu │ │ │ ├── binary.cuh │ │ │ ├── bitwise_binary.cu │ │ │ ├── divide.cu │ │ │ ├── equal.cu │ │ │ ├── greater.cu │ │ │ ├── greater_equal.cu │ │ │ ├── less.cu │ │ │ ├── less_equal.cu │ │ │ ├── log_add_exp.cu │ │ │ ├── logical_and.cu │ │ │ ├── logical_or.cu │ │ │ ├── maximum.cu │ │ │ ├── minimum.cu │ │ │ ├── multiply.cu │ │ │ ├── not_equal.cu │ │ │ ├── power.cu │ │ │ ├── remainder.cu │ │ │ └── subtract.cu │ │ ├── binary_two.cu │ │ ├── compiled.cpp │ │ ├── conv.cpp │ │ ├── conv │ │ │ ├── conv.h │ │ │ ├── gemm_conv.cu │ │ │ └── gemm_grouped_conv.cu │ │ ├── copy.cu │ │ ├── copy │ │ │ ├── copy.cuh │ │ │ ├── copy_contiguous.cu │ │ │ ├── copy_general.cu │ │ │ ├── copy_general_dynamic.cu │ │ │ └── copy_general_input.cu │ │ ├── cuda.cpp │ │ ├── cuda.h │ │ ├── cuda_utils.h │ │ ├── cudnn_utils.cpp │ │ ├── cudnn_utils.h │ │ ├── custom_kernel.cpp │ │ ├── device.cpp │ │ ├── device.h │ │ ├── device │ │ │ ├── atomic_ops.cuh │ │ │ ├── binary_ops.cuh │ │ │ ├── cast_op.cuh │ │ │ ├── complex.cuh │ │ │ ├── config.h │ │ │ ├── fp16_math.cuh │ │ │ ├── gather.cuh │ │ │ ├── gather_axis.cuh │ │ │ ├── indexing.cuh │ │ │ ├── scatter.cuh │ │ │ ├── scatter_axis.cuh │ │ │ ├── scatter_ops.cuh │ │ │ ├── ternary_ops.cuh │ │ │ ├── unary_ops.cuh │ │ │ └── utils.cuh │ │ ├── distributed.cu │ │ ├── eval.cpp │ │ ├── event.cu │ │ ├── event.h │ │ ├── fence.cpp │ │ ├── gemms │ │ │ ├── cublas_gemm.cpp │ │ │ ├── cublas_gemm.h │ │ │ ├── cublas_gemm_batched_12_0.cpp │ │ │ ├── cublas_gemm_batched_12_9.cu │ │ │ ├── gemv.cu │ │ │ └── gemv.h │ │ ├── indexing.cpp │ │ ├── jit_module.cpp │ │ ├── jit_module.h │ │ ├── kernel_utils.cu │ │ ├── kernel_utils.cuh │ │ ├── layer_norm.cu │ │ ├── load.cpp │ │ ├── logsumexp.cu │ │ ├── lru_cache.h │ │ ├── matmul.cpp │ │ ├── no_cuda.cpp │ │ ├── primitives.cpp │ │ ├── quantized │ │ │ ├── affine_quantize.cu │ │ │ ├── convert_fp8.cu │ │ │ ├── cuda_fp4.h │ │ │ ├── fp_quantize.cu │ │ │ ├── quantized.cpp │ │ │ ├── quantized.h │ │ │ └── quantized_utils.cuh │ │ ├── random.cu │ │ ├── reduce.cu │ │ ├── reduce │ │ │ ├── all_reduce.cu │ │ │ ├── col_reduce.cu │ │ │ ├── init_reduce.cu │ │ │ ├── reduce.cuh │ │ │ ├── reduce_ops.cuh │ │ │ ├── reduce_utils.cuh │ │ │ └── row_reduce.cu │ │ ├── rms_norm.cu │ │ ├── rope.cu │ │ ├── scaled_dot_product_attention.cpp │ │ ├── scaled_dot_product_attention.cu │ │ ├── scan.cu │ │ ├── slicing.cpp │ │ ├── softmax.cu │ │ ├── sort.cu │ │ ├── steel │ │ │ ├── defines.cuh │ │ │ ├── gemm.cuh │ │ │ ├── mma.cuh │ │ │ ├── tiles.cuh │ │ │ └── utils.cuh │ │ ├── ternary.cu │ │ ├── unary │ │ │ ├── CMakeLists.txt │ │ │ ├── abs.cu │ │ │ ├── arccos.cu │ │ │ ├── arccosh.cu │ │ │ ├── arcsin.cu │ │ │ ├── arcsinh.cu │ │ │ ├── arctan.cu │ │ │ ├── arctanh.cu │ │ │ ├── bitwise_invert.cu │ │ │ ├── ceil.cu │ │ │ ├── conjugate.cu │ │ │ ├── cos.cu │ │ │ ├── cosh.cu │ │ │ ├── erf.cu │ │ │ ├── erf_inv.cu │ │ │ ├── exp.cu │ │ │ ├── expm1.cu │ │ │ ├── floor.cu │ │ │ ├── imag.cu │ │ │ ├── log.cu │ │ │ ├── log1p.cu │ │ │ ├── logical_not.cu │ │ │ ├── negative.cu │ │ │ ├── real.cu │ │ │ ├── round.cu │ │ │ ├── sigmoid.cu │ │ │ ├── sign.cu │ │ │ ├── sin.cu │ │ │ ├── sinh.cu │ │ │ ├── sqrt.cu │ │ │ ├── square.cu │ │ │ ├── tan.cu │ │ │ ├── tanh.cu │ │ │ └── unary.cuh │ │ ├── utils.cpp │ │ ├── utils.h │ │ ├── worker.cpp │ │ └── worker.h │ ├── gpu │ │ ├── CMakeLists.txt │ │ ├── available.h │ │ ├── copy.cpp │ │ ├── copy.h │ │ ├── eval.h │ │ ├── primitives.cpp │ │ ├── slicing.cpp │ │ └── slicing.h │ ├── metal │ │ ├── CMakeLists.txt │ │ ├── allocator.cpp │ │ ├── allocator.h │ │ ├── binary.cpp │ │ ├── binary.h │ │ ├── compiled.cpp │ │ ├── conv.cpp │ │ ├── copy.cpp │ │ ├── custom_kernel.cpp │ │ ├── device.cpp │ │ ├── device.h │ │ ├── distributed.cpp │ │ ├── eval.cpp │ │ ├── event.cpp │ │ ├── fence.cpp │ │ ├── fft.cpp │ │ ├── hadamard.cpp │ │ ├── indexing.cpp │ │ ├── jit │ │ │ ├── includes.h │ │ │ └── indexing.h │ │ ├── jit_kernels.cpp │ │ ├── kernels.h │ │ ├── kernels │ │ │ ├── CMakeLists.txt │ │ │ ├── arange.h │ │ │ ├── arange.metal │ │ │ ├── arg_reduce.metal │ │ │ ├── atomic.h │ │ │ ├── bf16.h │ │ │ ├── bf16_math.h │ │ │ ├── binary.h │ │ │ ├── binary.metal │ │ │ ├── binary_ops.h │ │ │ ├── binary_two.h │ │ │ ├── binary_two.metal │ │ │ ├── cexpf.h │ │ │ ├── complex.h │ │ │ ├── conv.metal │ │ │ ├── copy.h │ │ │ ├── copy.metal │ │ │ ├── defines.h │ │ │ ├── erf.h │ │ │ ├── expm1f.h │ │ │ ├── fence.metal │ │ │ ├── fft.h │ │ │ ├── fft.metal │ │ │ ├── fft │ │ │ │ ├── radix.h │ │ │ │ └── readwrite.h │ │ │ ├── fp4.h │ │ │ ├── fp8.h │ │ │ ├── fp_quantized.h │ │ │ ├── fp_quantized.metal │ │ │ ├── fp_quantized_nax.h │ │ │ ├── fp_quantized_nax.metal │ │ │ ├── gemv.metal │ │ │ ├── gemv_masked.h │ │ │ ├── gemv_masked.metal │ │ │ ├── hadamard.h │ │ │ ├── indexing │ │ │ │ ├── gather.h │ │ │ │ ├── gather_axis.h │ │ │ │ ├── gather_front.h │ │ │ │ ├── indexing.h │ │ │ │ ├── masked_scatter.h │ │ │ │ ├── scatter.h │ │ │ │ └── scatter_axis.h │ │ │ ├── layer_norm.metal │ │ │ ├── logsumexp.h │ │ │ ├── logsumexp.metal │ │ │ ├── quantized.h │ │ │ ├── quantized.metal │ │ │ ├── quantized_nax.h │ │ │ ├── quantized_nax.metal │ │ │ ├── quantized_utils.h │ │ │ ├── random.metal │ │ │ ├── reduce.h │ │ │ ├── reduce.metal │ │ │ ├── reduce_utils.h │ │ │ ├── reduction │ │ │ │ ├── ops.h │ │ │ │ ├── reduce_all.h │ │ │ │ ├── reduce_col.h │ │ │ │ ├── reduce_init.h │ │ │ │ └── reduce_row.h │ │ │ ├── rms_norm.metal │ │ │ ├── rope.metal │ │ │ ├── scaled_dot_product_attention.metal │ │ │ ├── scan.h │ │ │ ├── scan.metal │ │ │ ├── sdpa_vector.h │ │ │ ├── softmax.h │ │ │ ├── softmax.metal │ │ │ ├── sort.h │ │ │ ├── sort.metal │ │ │ ├── steel │ │ │ │ ├── attn │ │ │ │ │ ├── attn.h │ │ │ │ │ ├── kernels │ │ │ │ │ │ ├── steel_attention.h │ │ │ │ │ │ ├── steel_attention.metal │ │ │ │ │ │ ├── steel_attention_nax.h │ │ │ │ │ │ └── steel_attention_nax.metal │ │ │ │ │ ├── loader.h │ │ │ │ │ ├── mma.h │ │ │ │ │ ├── nax.h │ │ │ │ │ ├── params.h │ │ │ │ │ └── transforms.h │ │ │ │ ├── conv │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── kernels │ │ │ │ │ │ ├── steel_conv.h │ │ │ │ │ │ ├── steel_conv.metal │ │ │ │ │ │ ├── steel_conv_general.h │ │ │ │ │ │ └── steel_conv_general.metal │ │ │ │ │ ├── loader.h │ │ │ │ │ ├── loaders │ │ │ │ │ │ ├── loader_channel_l.h │ │ │ │ │ │ ├── loader_channel_n.h │ │ │ │ │ │ └── loader_general.h │ │ │ │ │ └── params.h │ │ │ │ ├── defines.h │ │ │ │ ├── gemm │ │ │ │ │ ├── gemm.h │ │ │ │ │ ├── gemm_nax.h │ │ │ │ │ ├── kernels │ │ │ │ │ │ ├── steel_gemm_fused.h │ │ │ │ │ │ ├── steel_gemm_fused.metal │ │ │ │ │ │ ├── steel_gemm_fused_nax.h │ │ │ │ │ │ ├── steel_gemm_fused_nax.metal │ │ │ │ │ │ ├── steel_gemm_gather.h │ │ │ │ │ │ ├── steel_gemm_gather.metal │ │ │ │ │ │ ├── steel_gemm_gather_nax.h │ │ │ │ │ │ ├── steel_gemm_gather_nax.metal │ │ │ │ │ │ ├── steel_gemm_masked.h │ │ │ │ │ │ ├── steel_gemm_masked.metal │ │ │ │ │ │ ├── steel_gemm_segmented.h │ │ │ │ │ │ ├── steel_gemm_segmented.metal │ │ │ │ │ │ ├── steel_gemm_splitk.h │ │ │ │ │ │ └── steel_gemm_splitk.metal │ │ │ │ │ ├── loader.h │ │ │ │ │ ├── mma.h │ │ │ │ │ ├── nax.h │ │ │ │ │ ├── params.h │ │ │ │ │ └── transforms.h │ │ │ │ ├── utils.h │ │ │ │ └── utils │ │ │ │ │ ├── integral_constant.h │ │ │ │ │ └── type_traits.h │ │ │ ├── ternary.h │ │ │ ├── ternary.metal │ │ │ ├── ternary_ops.h │ │ │ ├── unary.h │ │ │ ├── unary.metal │ │ │ ├── unary_ops.h │ │ │ └── utils.h │ │ ├── logsumexp.cpp │ │ ├── make_compiled_preamble.sh │ │ ├── matmul.cpp │ │ ├── matmul.h │ │ ├── metal.cpp │ │ ├── metal.h │ │ ├── no_metal.cpp │ │ ├── nojit_kernels.cpp │ │ ├── normalization.cpp │ │ ├── primitives.cpp │ │ ├── quantized.cpp │ │ ├── reduce.cpp │ │ ├── reduce.h │ │ ├── resident.cpp │ │ ├── resident.h │ │ ├── rope.cpp │ │ ├── scaled_dot_product_attention.cpp │ │ ├── scan.cpp │ │ ├── scan.h │ │ ├── slicing.cpp │ │ ├── softmax.cpp │ │ ├── sort.cpp │ │ ├── ternary.cpp │ │ ├── ternary.h │ │ ├── unary.cpp │ │ ├── unary.h │ │ ├── utils.cpp │ │ └── utils.h │ ├── no_cpu │ │ ├── CMakeLists.txt │ │ ├── available.cpp │ │ ├── compiled.cpp │ │ └── primitives.cpp │ └── no_gpu │ │ ├── CMakeLists.txt │ │ ├── allocator.cpp │ │ ├── apple_memory.h │ │ ├── eval.cpp │ │ ├── event.cpp │ │ ├── fence.cpp │ │ ├── linux_memory.h │ │ └── primitives.cpp ├── compile.cpp ├── compile.h ├── compile_impl.h ├── device.cpp ├── device.h ├── distributed │ ├── CMakeLists.txt │ ├── distributed.cpp │ ├── distributed.h │ ├── distributed_impl.h │ ├── mpi │ │ ├── CMakeLists.txt │ │ ├── mpi.cpp │ │ ├── mpi.h │ │ ├── mpi_declarations.h │ │ └── no_mpi.cpp │ ├── nccl │ │ ├── CMakeLists.txt │ │ ├── nccl.cpp │ │ ├── nccl.h │ │ ├── nccl_stub │ │ │ ├── CMakeLists.txt │ │ │ └── nccl_stubs.cpp │ │ └── no_nccl.cpp │ ├── ops.cpp │ ├── ops.h │ ├── primitives.cpp │ ├── primitives.h │ └── ring │ │ ├── CMakeLists.txt │ │ ├── no_ring.cpp │ │ ├── ring.cpp │ │ └── ring.h ├── dtype.cpp ├── dtype.h ├── dtype_utils.cpp ├── dtype_utils.h ├── einsum.cpp ├── einsum.h ├── event.h ├── export.cpp ├── export.h ├── export_impl.h ├── fast.cpp ├── fast.h ├── fast_primitives.h ├── fence.h ├── fft.cpp ├── fft.h ├── graph_utils.cpp ├── graph_utils.h ├── io.h ├── io │ ├── CMakeLists.txt │ ├── gguf.cpp │ ├── gguf.h │ ├── gguf_quants.cpp │ ├── load.cpp │ ├── load.h │ ├── no_gguf.cpp │ ├── no_safetensors.cpp │ └── safetensors.cpp ├── linalg.cpp ├── linalg.h ├── memory.h ├── mlx.h ├── ops.cpp ├── ops.h ├── primitives.cpp ├── primitives.h ├── random.cpp ├── random.h ├── scheduler.cpp ├── scheduler.h ├── small_vector.h ├── stream.h ├── threadpool.h ├── transforms.cpp ├── transforms.h ├── transforms_impl.h ├── types │ ├── bf16.h │ ├── complex.h │ ├── fp16.h │ ├── half_types.h │ └── limits.h ├── utils.cpp ├── utils.h ├── version.cpp └── version.h ├── pyproject.toml ├── python ├── mlx │ ├── __main__.py │ ├── _reprlib_fix.py │ ├── _stub_patterns.txt │ ├── distributed_run.py │ ├── extension.py │ ├── nn │ │ ├── __init__.py │ │ ├── init.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── activations.py │ │ │ ├── base.py │ │ │ ├── containers.py │ │ │ ├── convolution.py │ │ │ ├── convolution_transpose.py │ │ │ ├── distributed.py │ │ │ ├── dropout.py │ │ │ ├── embedding.py │ │ │ ├── linear.py │ │ │ ├── normalization.py │ │ │ ├── pooling.py │ │ │ ├── positional_encoding.py │ │ │ ├── quantized.py │ │ │ ├── recurrent.py │ │ │ ├── transformer.py │ │ │ └── upsample.py │ │ ├── losses.py │ │ └── utils.py │ ├── optimizers │ │ ├── __init__.py │ │ ├── optimizers.py │ │ └── schedulers.py │ ├── py.typed │ └── utils.py ├── scripts │ ├── repair_cuda.sh │ ├── repair_linux.sh │ └── repair_record.py ├── src │ ├── CMakeLists.txt │ ├── array.cpp │ ├── buffer.h │ ├── constants.cpp │ ├── convert.cpp │ ├── convert.h │ ├── cuda.cpp │ ├── device.cpp │ ├── distributed.cpp │ ├── export.cpp │ ├── fast.cpp │ ├── fft.cpp │ ├── indexing.cpp │ ├── indexing.h │ ├── linalg.cpp │ ├── load.cpp │ ├── load.h │ ├── memory.cpp │ ├── metal.cpp │ ├── mlx.cpp │ ├── mlx_func.cpp │ ├── mlx_func.h │ ├── ops.cpp │ ├── random.cpp │ ├── small_vector.h │ ├── stream.cpp │ ├── transforms.cpp │ ├── trees.cpp │ ├── trees.h │ ├── utils.cpp │ └── utils.h └── tests │ ├── __main__.py │ ├── cuda_skip.py │ ├── mlx_distributed_tests.py │ ├── mlx_tests.py │ ├── mpi_test_distributed.py │ ├── nccl_test_distributed.py │ ├── ring_test_distributed.py │ ├── test_array.py │ ├── test_autograd.py │ ├── test_bf16.py │ ├── test_blas.py │ ├── test_compile.py │ ├── test_constants.py │ ├── test_conv.py │ ├── test_conv_transpose.py │ ├── test_device.py │ ├── test_double.py │ ├── test_einsum.py │ ├── test_eval.py │ ├── test_export_import.py │ ├── test_fast.py │ ├── test_fast_sdpa.py │ ├── test_fft.py │ ├── test_graph.py │ ├── test_init.py │ ├── test_linalg.py │ ├── test_load.py │ ├── test_losses.py │ ├── test_memory.py │ ├── test_nn.py │ ├── test_ops.py │ ├── test_optimizers.py │ ├── test_quantized.py │ ├── test_random.py │ ├── test_reduce.py │ ├── test_tree.py │ ├── test_upsample.py │ └── test_vmap.py ├── setup.py └── tests ├── CMakeLists.txt ├── allocator_tests.cpp ├── arg_reduce_tests.cpp ├── array_tests.cpp ├── autograd_tests.cpp ├── blas_tests.cpp ├── compile_tests.cpp ├── creations_tests.cpp ├── custom_vjp_tests.cpp ├── device_tests.cpp ├── einsum_tests.cpp ├── eval_tests.cpp ├── export_import_tests.cpp ├── fft_tests.cpp ├── gpu_tests.cpp ├── linalg_tests.cpp ├── load_tests.cpp ├── ops_tests.cpp ├── random_tests.cpp ├── scheduler_tests.cpp ├── tests.cpp ├── utils_tests.cpp └── vmap_tests.cpp /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.clang-format -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/actions/build-cuda-release/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-cuda-release/action.yml -------------------------------------------------------------------------------- /.github/actions/build-docs/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-docs/action.yml -------------------------------------------------------------------------------- /.github/actions/build-linux-release/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-linux-release/action.yml -------------------------------------------------------------------------------- /.github/actions/build-linux/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-linux/action.yml -------------------------------------------------------------------------------- /.github/actions/build-macos-release/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-macos-release/action.yml -------------------------------------------------------------------------------- /.github/actions/build-macos/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/build-macos/action.yml -------------------------------------------------------------------------------- /.github/actions/setup-linux/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/setup-linux/action.yml -------------------------------------------------------------------------------- /.github/actions/setup-macos/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/setup-macos/action.yml -------------------------------------------------------------------------------- /.github/actions/test-linux/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/actions/test-linux/action.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/scripts/setup+build-cpp-linux-fedora-container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/scripts/setup+build-cpp-linux-fedora-container.sh -------------------------------------------------------------------------------- /.github/workflows/build_and_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/workflows/build_and_test.yml -------------------------------------------------------------------------------- /.github/workflows/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/workflows/documentation.yml -------------------------------------------------------------------------------- /.github/workflows/nightly.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/workflows/nightly.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /ACKNOWLEDGMENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/ACKNOWLEDGMENTS.md -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /benchmarks/cpp/autograd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/autograd.cpp -------------------------------------------------------------------------------- /benchmarks/cpp/compare_devices.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/compare_devices.cpp -------------------------------------------------------------------------------- /benchmarks/cpp/irregular_strides.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/irregular_strides.cpp -------------------------------------------------------------------------------- /benchmarks/cpp/single_ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/single_ops.cpp -------------------------------------------------------------------------------- /benchmarks/cpp/time_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/cpp/time_utils.h -------------------------------------------------------------------------------- /benchmarks/numpy/single_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/numpy/single_ops.py -------------------------------------------------------------------------------- /benchmarks/numpy/time_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/numpy/time_utils.py -------------------------------------------------------------------------------- /benchmarks/python/batch_matmul_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/batch_matmul_bench.py -------------------------------------------------------------------------------- /benchmarks/python/blas/bench_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/blas/bench_gemm.py -------------------------------------------------------------------------------- /benchmarks/python/blas/bench_gemv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/blas/bench_gemv.py -------------------------------------------------------------------------------- /benchmarks/python/comparative/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/comparative/README.md -------------------------------------------------------------------------------- /benchmarks/python/comparative/bench_mlx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/comparative/bench_mlx.py -------------------------------------------------------------------------------- /benchmarks/python/comparative/bench_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/comparative/bench_torch.py -------------------------------------------------------------------------------- /benchmarks/python/comparative/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/comparative/compare.py -------------------------------------------------------------------------------- /benchmarks/python/compile_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/compile_bench.py -------------------------------------------------------------------------------- /benchmarks/python/conv1d_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv1d_bench.py -------------------------------------------------------------------------------- /benchmarks/python/conv2d_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv2d_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv2d_train_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv2d_train_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv2d_transpose_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv2d_transpose_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv3d_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv3d_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv3d_train_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv3d_train_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv3d_transpose_bench_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv3d_transpose_bench_cpu.py -------------------------------------------------------------------------------- /benchmarks/python/conv_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv_bench.py -------------------------------------------------------------------------------- /benchmarks/python/conv_transpose_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv_transpose_bench.py -------------------------------------------------------------------------------- /benchmarks/python/conv_unaligned_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/conv_unaligned_bench.py -------------------------------------------------------------------------------- /benchmarks/python/distributed_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/distributed_bench.py -------------------------------------------------------------------------------- /benchmarks/python/einsum_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/einsum_bench.py -------------------------------------------------------------------------------- /benchmarks/python/fft_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/fft_bench.py -------------------------------------------------------------------------------- /benchmarks/python/gather_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/gather_bench.py -------------------------------------------------------------------------------- /benchmarks/python/gather_mm_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/gather_mm_bench.py -------------------------------------------------------------------------------- /benchmarks/python/gather_qmm_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/gather_qmm_bench.py -------------------------------------------------------------------------------- /benchmarks/python/hadamard_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/hadamard_bench.py -------------------------------------------------------------------------------- /benchmarks/python/layer_norm_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/layer_norm_bench.py -------------------------------------------------------------------------------- /benchmarks/python/masked_scatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/masked_scatter.py -------------------------------------------------------------------------------- /benchmarks/python/rms_norm_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/rms_norm_bench.py -------------------------------------------------------------------------------- /benchmarks/python/rope_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/rope_bench.py -------------------------------------------------------------------------------- /benchmarks/python/scatter_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/scatter_bench.py -------------------------------------------------------------------------------- /benchmarks/python/sdpa_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/sdpa_bench.py -------------------------------------------------------------------------------- /benchmarks/python/sdpa_vector_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/sdpa_vector_bench.py -------------------------------------------------------------------------------- /benchmarks/python/single_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/single_ops.py -------------------------------------------------------------------------------- /benchmarks/python/synchronize_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/synchronize_bench.py -------------------------------------------------------------------------------- /benchmarks/python/time_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/benchmarks/python/time_utils.py -------------------------------------------------------------------------------- /cmake/FindNCCL.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/cmake/FindNCCL.cmake -------------------------------------------------------------------------------- /cmake/Findnvpl.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/cmake/Findnvpl.cmake -------------------------------------------------------------------------------- /cmake/extension.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/cmake/extension.cmake -------------------------------------------------------------------------------- /docs/.clang-format: -------------------------------------------------------------------------------- 1 | DisableFormat: true 2 | SortIncludes: Never 3 | -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/.gitignore -------------------------------------------------------------------------------- /docs/.nojekyll: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/Doxyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/Doxyfile -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/src/_static/metal_debugger/capture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_static/metal_debugger/capture.png -------------------------------------------------------------------------------- /docs/src/_static/metal_debugger/schema.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_static/metal_debugger/schema.png -------------------------------------------------------------------------------- /docs/src/_static/mlx_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_static/mlx_logo.png -------------------------------------------------------------------------------- /docs/src/_static/mlx_logo_dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_static/mlx_logo_dark.png -------------------------------------------------------------------------------- /docs/src/_templates/module-base-class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_templates/module-base-class.rst -------------------------------------------------------------------------------- /docs/src/_templates/nn-module-template.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_templates/nn-module-template.rst -------------------------------------------------------------------------------- /docs/src/_templates/optimizers-template.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/_templates/optimizers-template.rst -------------------------------------------------------------------------------- /docs/src/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/conf.py -------------------------------------------------------------------------------- /docs/src/cpp/ops.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/cpp/ops.rst -------------------------------------------------------------------------------- /docs/src/dev/custom_metal_kernels.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/dev/custom_metal_kernels.rst -------------------------------------------------------------------------------- /docs/src/dev/extensions.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/dev/extensions.rst -------------------------------------------------------------------------------- /docs/src/dev/metal_debugger.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/dev/metal_debugger.rst -------------------------------------------------------------------------------- /docs/src/dev/mlx_in_cpp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/dev/mlx_in_cpp.rst -------------------------------------------------------------------------------- /docs/src/examples/linear_regression.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/examples/linear_regression.rst -------------------------------------------------------------------------------- /docs/src/examples/llama-inference.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/examples/llama-inference.rst -------------------------------------------------------------------------------- /docs/src/examples/mlp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/examples/mlp.rst -------------------------------------------------------------------------------- /docs/src/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/index.rst -------------------------------------------------------------------------------- /docs/src/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/install.rst -------------------------------------------------------------------------------- /docs/src/python/array.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/array.rst -------------------------------------------------------------------------------- /docs/src/python/cuda.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/cuda.rst -------------------------------------------------------------------------------- /docs/src/python/data_types.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/data_types.rst -------------------------------------------------------------------------------- /docs/src/python/devices_and_streams.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/devices_and_streams.rst -------------------------------------------------------------------------------- /docs/src/python/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/distributed.rst -------------------------------------------------------------------------------- /docs/src/python/export.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/export.rst -------------------------------------------------------------------------------- /docs/src/python/fast.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/fast.rst -------------------------------------------------------------------------------- /docs/src/python/fft.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/fft.rst -------------------------------------------------------------------------------- /docs/src/python/linalg.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/linalg.rst -------------------------------------------------------------------------------- /docs/src/python/memory_management.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/memory_management.rst -------------------------------------------------------------------------------- /docs/src/python/metal.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/metal.rst -------------------------------------------------------------------------------- /docs/src/python/nn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn.rst -------------------------------------------------------------------------------- /docs/src/python/nn/functions.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn/functions.rst -------------------------------------------------------------------------------- /docs/src/python/nn/init.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn/init.rst -------------------------------------------------------------------------------- /docs/src/python/nn/layers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn/layers.rst -------------------------------------------------------------------------------- /docs/src/python/nn/losses.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn/losses.rst -------------------------------------------------------------------------------- /docs/src/python/nn/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/nn/module.rst -------------------------------------------------------------------------------- /docs/src/python/ops.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/ops.rst -------------------------------------------------------------------------------- /docs/src/python/optimizers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/optimizers.rst -------------------------------------------------------------------------------- /docs/src/python/optimizers/common_optimizers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/optimizers/common_optimizers.rst -------------------------------------------------------------------------------- /docs/src/python/optimizers/optimizer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/optimizers/optimizer.rst -------------------------------------------------------------------------------- /docs/src/python/optimizers/schedulers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/optimizers/schedulers.rst -------------------------------------------------------------------------------- /docs/src/python/random.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/random.rst -------------------------------------------------------------------------------- /docs/src/python/transforms.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/transforms.rst -------------------------------------------------------------------------------- /docs/src/python/tree_utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/python/tree_utils.rst -------------------------------------------------------------------------------- /docs/src/usage/compile.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/compile.rst -------------------------------------------------------------------------------- /docs/src/usage/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/distributed.rst -------------------------------------------------------------------------------- /docs/src/usage/export.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/export.rst -------------------------------------------------------------------------------- /docs/src/usage/function_transforms.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/function_transforms.rst -------------------------------------------------------------------------------- /docs/src/usage/indexing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/indexing.rst -------------------------------------------------------------------------------- /docs/src/usage/launching_distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/launching_distributed.rst -------------------------------------------------------------------------------- /docs/src/usage/lazy_evaluation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/lazy_evaluation.rst -------------------------------------------------------------------------------- /docs/src/usage/numpy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/numpy.rst -------------------------------------------------------------------------------- /docs/src/usage/quick_start.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/quick_start.rst -------------------------------------------------------------------------------- /docs/src/usage/saving_and_loading.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/saving_and_loading.rst -------------------------------------------------------------------------------- /docs/src/usage/unified_memory.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/unified_memory.rst -------------------------------------------------------------------------------- /docs/src/usage/using_streams.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/docs/src/usage/using_streams.rst -------------------------------------------------------------------------------- /examples/cmake_project/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cmake_project/CMakeLists.txt -------------------------------------------------------------------------------- /examples/cmake_project/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cmake_project/README.md -------------------------------------------------------------------------------- /examples/cmake_project/example.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cmake_project/example.cpp -------------------------------------------------------------------------------- /examples/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /examples/cpp/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/distributed.cpp -------------------------------------------------------------------------------- /examples/cpp/linear_regression.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/linear_regression.cpp -------------------------------------------------------------------------------- /examples/cpp/logistic_regression.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/logistic_regression.cpp -------------------------------------------------------------------------------- /examples/cpp/metal_capture.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/metal_capture.cpp -------------------------------------------------------------------------------- /examples/cpp/timer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/timer.h -------------------------------------------------------------------------------- /examples/cpp/tutorial.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/cpp/tutorial.cpp -------------------------------------------------------------------------------- /examples/export/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/CMakeLists.txt -------------------------------------------------------------------------------- /examples/export/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/README.md -------------------------------------------------------------------------------- /examples/export/eval_mlp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/eval_mlp.cpp -------------------------------------------------------------------------------- /examples/export/eval_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/eval_mlp.py -------------------------------------------------------------------------------- /examples/export/train_mlp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/train_mlp.cpp -------------------------------------------------------------------------------- /examples/export/train_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/export/train_mlp.py -------------------------------------------------------------------------------- /examples/extensions/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/CMakeLists.txt -------------------------------------------------------------------------------- /examples/extensions/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/README.md -------------------------------------------------------------------------------- /examples/extensions/axpby/axpby.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/axpby/axpby.cpp -------------------------------------------------------------------------------- /examples/extensions/axpby/axpby.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/axpby/axpby.h -------------------------------------------------------------------------------- /examples/extensions/axpby/axpby.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/axpby/axpby.metal -------------------------------------------------------------------------------- /examples/extensions/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/bindings.cpp -------------------------------------------------------------------------------- /examples/extensions/mlx_sample_extensions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/mlx_sample_extensions/__init__.py -------------------------------------------------------------------------------- /examples/extensions/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/pyproject.toml -------------------------------------------------------------------------------- /examples/extensions/requirements.txt: -------------------------------------------------------------------------------- 1 | setuptools>=42 2 | cmake>=3.25 3 | mlx>=0.21.0 4 | nanobind==2.4.0 5 | -------------------------------------------------------------------------------- /examples/extensions/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/setup.py -------------------------------------------------------------------------------- /examples/extensions/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/extensions/test.py -------------------------------------------------------------------------------- /examples/python/linear_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/python/linear_regression.py -------------------------------------------------------------------------------- /examples/python/logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/examples/python/logistic_regression.py -------------------------------------------------------------------------------- /mlx.pc.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx.pc.in -------------------------------------------------------------------------------- /mlx/3rdparty/.clang-format: -------------------------------------------------------------------------------- 1 | DisableFormat: true 2 | SortIncludes: Never 3 | -------------------------------------------------------------------------------- /mlx/3rdparty/pocketfft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/3rdparty/pocketfft.h -------------------------------------------------------------------------------- /mlx/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/allocator.h -------------------------------------------------------------------------------- /mlx/array.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/array.cpp -------------------------------------------------------------------------------- /mlx/array.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/array.h -------------------------------------------------------------------------------- /mlx/backend/common/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/common/binary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/binary.h -------------------------------------------------------------------------------- /mlx/backend/common/broadcasting.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/broadcasting.cpp -------------------------------------------------------------------------------- /mlx/backend/common/broadcasting.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/broadcasting.h -------------------------------------------------------------------------------- /mlx/backend/common/buffer_cache.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/buffer_cache.h -------------------------------------------------------------------------------- /mlx/backend/common/common.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/common.cpp -------------------------------------------------------------------------------- /mlx/backend/common/compiled.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/compiled.cpp -------------------------------------------------------------------------------- /mlx/backend/common/compiled.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/compiled.h -------------------------------------------------------------------------------- /mlx/backend/common/copy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/copy.h -------------------------------------------------------------------------------- /mlx/backend/common/hadamard.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/hadamard.h -------------------------------------------------------------------------------- /mlx/backend/common/load.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/load.cpp -------------------------------------------------------------------------------- /mlx/backend/common/matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/matmul.h -------------------------------------------------------------------------------- /mlx/backend/common/reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/reduce.cpp -------------------------------------------------------------------------------- /mlx/backend/common/reduce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/reduce.h -------------------------------------------------------------------------------- /mlx/backend/common/slicing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/slicing.cpp -------------------------------------------------------------------------------- /mlx/backend/common/slicing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/slicing.h -------------------------------------------------------------------------------- /mlx/backend/common/ternary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/ternary.h -------------------------------------------------------------------------------- /mlx/backend/common/unary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/unary.h -------------------------------------------------------------------------------- /mlx/backend/common/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/utils.cpp -------------------------------------------------------------------------------- /mlx/backend/common/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/common/utils.h -------------------------------------------------------------------------------- /mlx/backend/cpu/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/cpu/arange.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/arange.h -------------------------------------------------------------------------------- /mlx/backend/cpu/arg_reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/arg_reduce.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/available.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/available.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/available.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/available.h -------------------------------------------------------------------------------- /mlx/backend/cpu/binary.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/binary.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/binary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/binary.h -------------------------------------------------------------------------------- /mlx/backend/cpu/binary_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/binary_ops.h -------------------------------------------------------------------------------- /mlx/backend/cpu/binary_two.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/binary_two.h -------------------------------------------------------------------------------- /mlx/backend/cpu/cholesky.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/cholesky.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/compiled.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/compiled.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/compiled_preamble.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/compiled_preamble.h -------------------------------------------------------------------------------- /mlx/backend/cpu/conv.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/conv.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/copy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/copy.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/copy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/copy.h -------------------------------------------------------------------------------- /mlx/backend/cpu/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/distributed.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/eig.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/eig.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/eigh.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/eigh.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/encoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/encoder.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/encoder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/encoder.h -------------------------------------------------------------------------------- /mlx/backend/cpu/eval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/eval.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/eval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/eval.h -------------------------------------------------------------------------------- /mlx/backend/cpu/fft.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/fft.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemm.h -------------------------------------------------------------------------------- /mlx/backend/cpu/gemms/bnns.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemms/bnns.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/gemms/cblas.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemms/cblas.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/gemms/simd_bf16.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemms/simd_bf16.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/gemms/simd_fp16.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemms/simd_fp16.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/gemms/simd_gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/gemms/simd_gemm.h -------------------------------------------------------------------------------- /mlx/backend/cpu/hadamard.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/hadamard.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/indexing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/indexing.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/inverse.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/inverse.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/jit_compiler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/jit_compiler.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/jit_compiler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/jit_compiler.h -------------------------------------------------------------------------------- /mlx/backend/cpu/lapack.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/lapack.h -------------------------------------------------------------------------------- /mlx/backend/cpu/logsumexp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/logsumexp.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/luf.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/luf.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/make_compiled_preamble.ps1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/make_compiled_preamble.ps1 -------------------------------------------------------------------------------- /mlx/backend/cpu/make_compiled_preamble.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/make_compiled_preamble.sh -------------------------------------------------------------------------------- /mlx/backend/cpu/masked_mm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/masked_mm.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/matmul.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/primitives.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/qrf.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/qrf.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/quantized.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/quantized.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/reduce.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/scan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/scan.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/select.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/select.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/accelerate_fp16_simd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/accelerate_fp16_simd.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/accelerate_simd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/accelerate_simd.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/base_simd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/base_simd.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/math.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/math.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/neon_fp16_simd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/neon_fp16_simd.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/simd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/simd.h -------------------------------------------------------------------------------- /mlx/backend/cpu/simd/type.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/simd/type.h -------------------------------------------------------------------------------- /mlx/backend/cpu/slicing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/slicing.h -------------------------------------------------------------------------------- /mlx/backend/cpu/softmax.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/softmax.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/sort.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/sort.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/svd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/svd.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/ternary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/ternary.h -------------------------------------------------------------------------------- /mlx/backend/cpu/threefry.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/threefry.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/threefry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/threefry.h -------------------------------------------------------------------------------- /mlx/backend/cpu/unary.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/unary.cpp -------------------------------------------------------------------------------- /mlx/backend/cpu/unary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/unary.h -------------------------------------------------------------------------------- /mlx/backend/cpu/unary_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cpu/unary_ops.h -------------------------------------------------------------------------------- /mlx/backend/cuda/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/cuda/allocator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/allocator.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/allocator.h -------------------------------------------------------------------------------- /mlx/backend/cuda/arange.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/arange.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/arg_reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/arg_reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/bin2h.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/bin2h.cmake -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/add.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/add.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/arctan2.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/arctan2.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/binary.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/binary.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/bitwise_binary.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/bitwise_binary.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/divide.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/divide.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/equal.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/equal.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/greater.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/greater.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/greater_equal.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/greater_equal.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/less.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/less.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/less_equal.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/less_equal.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/log_add_exp.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/log_add_exp.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/logical_and.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/logical_and.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/logical_or.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/logical_or.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/maximum.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/maximum.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/minimum.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/minimum.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/multiply.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/multiply.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/not_equal.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/not_equal.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/power.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/power.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/remainder.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/remainder.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary/subtract.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary/subtract.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/binary_two.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/binary_two.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/compiled.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/compiled.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/conv.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/conv.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/conv/conv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/conv/conv.h -------------------------------------------------------------------------------- /mlx/backend/cuda/conv/gemm_conv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/conv/gemm_conv.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/conv/gemm_grouped_conv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/conv/gemm_grouped_conv.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/copy.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/copy/copy.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy/copy.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/copy/copy_contiguous.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy/copy_contiguous.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/copy/copy_general.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy/copy_general.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/copy/copy_general_dynamic.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy/copy_general_dynamic.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/copy/copy_general_input.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/copy/copy_general_input.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/cuda.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/cuda.h -------------------------------------------------------------------------------- /mlx/backend/cuda/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/cuda_utils.h -------------------------------------------------------------------------------- /mlx/backend/cuda/cudnn_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/cudnn_utils.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/cudnn_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/cudnn_utils.h -------------------------------------------------------------------------------- /mlx/backend/cuda/custom_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/custom_kernel.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/device.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device.h -------------------------------------------------------------------------------- /mlx/backend/cuda/device/atomic_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/atomic_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/binary_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/binary_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/cast_op.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/cast_op.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/complex.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/complex.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/config.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/config.h -------------------------------------------------------------------------------- /mlx/backend/cuda/device/fp16_math.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/fp16_math.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/gather.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/gather.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/gather_axis.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/gather_axis.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/indexing.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/indexing.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/scatter.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/scatter.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/scatter_axis.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/scatter_axis.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/scatter_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/scatter_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/ternary_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/ternary_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/unary_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/unary_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/device/utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/device/utils.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/distributed.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/distributed.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/eval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/eval.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/event.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/event.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/event.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/event.h -------------------------------------------------------------------------------- /mlx/backend/cuda/fence.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/fence.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/cublas_gemm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/cublas_gemm.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/cublas_gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/cublas_gemm.h -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/cublas_gemm_batched_12_0.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/cublas_gemm_batched_12_0.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/cublas_gemm_batched_12_9.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/cublas_gemm_batched_12_9.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/gemv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/gemv.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/gemms/gemv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/gemms/gemv.h -------------------------------------------------------------------------------- /mlx/backend/cuda/indexing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/indexing.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/jit_module.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/jit_module.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/jit_module.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/jit_module.h -------------------------------------------------------------------------------- /mlx/backend/cuda/kernel_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/kernel_utils.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/kernel_utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/kernel_utils.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/layer_norm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/layer_norm.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/load.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/load.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/logsumexp.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/logsumexp.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/lru_cache.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/lru_cache.h -------------------------------------------------------------------------------- /mlx/backend/cuda/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/matmul.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/no_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/no_cuda.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/primitives.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/affine_quantize.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/affine_quantize.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/convert_fp8.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/convert_fp8.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/cuda_fp4.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/cuda_fp4.h -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/fp_quantize.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/fp_quantize.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/quantized.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/quantized.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/quantized.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/quantized.h -------------------------------------------------------------------------------- /mlx/backend/cuda/quantized/quantized_utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/quantized/quantized_utils.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/random.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/random.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/all_reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/all_reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/col_reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/col_reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/init_reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/init_reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/reduce.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/reduce.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/reduce_ops.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/reduce_ops.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/reduce_utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/reduce_utils.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/reduce/row_reduce.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/reduce/row_reduce.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/rms_norm.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/rms_norm.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/rope.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/rope.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/scaled_dot_product_attention.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/scaled_dot_product_attention.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/scaled_dot_product_attention.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/scaled_dot_product_attention.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/scan.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/scan.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/slicing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/slicing.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/softmax.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/softmax.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/sort.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/sort.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/steel/defines.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/steel/defines.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/steel/gemm.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/steel/gemm.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/steel/mma.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/steel/mma.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/steel/tiles.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/steel/tiles.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/steel/utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/steel/utils.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/ternary.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/ternary.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/abs.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/abs.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arccos.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arccos.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arccosh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arccosh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arcsin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arcsin.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arcsinh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arcsinh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arctan.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arctan.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/arctanh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/arctanh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/bitwise_invert.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/bitwise_invert.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/ceil.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/ceil.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/conjugate.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/conjugate.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/cos.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/cos.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/cosh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/cosh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/erf.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/erf.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/erf_inv.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/erf_inv.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/exp.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/exp.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/expm1.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/expm1.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/floor.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/floor.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/imag.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/imag.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/log.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/log.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/log1p.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/log1p.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/logical_not.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/logical_not.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/negative.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/negative.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/real.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/real.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/round.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/round.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/sigmoid.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/sigmoid.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/sign.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/sign.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/sin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/sin.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/sinh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/sinh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/sqrt.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/sqrt.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/square.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/square.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/tan.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/tan.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/tanh.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/tanh.cu -------------------------------------------------------------------------------- /mlx/backend/cuda/unary/unary.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/unary/unary.cuh -------------------------------------------------------------------------------- /mlx/backend/cuda/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/utils.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/utils.h -------------------------------------------------------------------------------- /mlx/backend/cuda/worker.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/worker.cpp -------------------------------------------------------------------------------- /mlx/backend/cuda/worker.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/cuda/worker.h -------------------------------------------------------------------------------- /mlx/backend/gpu/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/gpu/available.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/available.h -------------------------------------------------------------------------------- /mlx/backend/gpu/copy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/copy.cpp -------------------------------------------------------------------------------- /mlx/backend/gpu/copy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/copy.h -------------------------------------------------------------------------------- /mlx/backend/gpu/eval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/eval.h -------------------------------------------------------------------------------- /mlx/backend/gpu/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/primitives.cpp -------------------------------------------------------------------------------- /mlx/backend/gpu/slicing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/slicing.cpp -------------------------------------------------------------------------------- /mlx/backend/gpu/slicing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/gpu/slicing.h -------------------------------------------------------------------------------- /mlx/backend/metal/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/metal/allocator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/allocator.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/allocator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/allocator.h -------------------------------------------------------------------------------- /mlx/backend/metal/binary.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/binary.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/binary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/binary.h -------------------------------------------------------------------------------- /mlx/backend/metal/compiled.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/compiled.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/conv.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/conv.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/copy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/copy.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/custom_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/custom_kernel.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/device.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/device.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/device.h -------------------------------------------------------------------------------- /mlx/backend/metal/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/distributed.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/eval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/eval.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/event.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/event.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/fence.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/fence.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/fft.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/fft.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/hadamard.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/hadamard.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/indexing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/indexing.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/jit/includes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/jit/includes.h -------------------------------------------------------------------------------- /mlx/backend/metal/jit/indexing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/jit/indexing.h -------------------------------------------------------------------------------- /mlx/backend/metal/jit_kernels.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/jit_kernels.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/arange.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/arange.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/arange.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/arange.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/arg_reduce.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/arg_reduce.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/atomic.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/atomic.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/bf16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/bf16.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/bf16_math.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/bf16_math.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/binary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/binary.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/binary.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/binary.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/binary_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/binary_ops.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/binary_two.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/binary_two.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/binary_two.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/binary_two.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/cexpf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/cexpf.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/complex.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/conv.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/conv.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/copy.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/copy.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/copy.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/copy.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/defines.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/defines.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/erf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/erf.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/expm1f.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/expm1f.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fence.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fence.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fft.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fft.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fft.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fft/radix.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fft/radix.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fft/readwrite.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fft/readwrite.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp4.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp4.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp8.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp8.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp_quantized.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp_quantized.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp_quantized.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp_quantized.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp_quantized_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp_quantized_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/fp_quantized_nax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/fp_quantized_nax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/gemv.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/gemv.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/gemv_masked.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/gemv_masked.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/gemv_masked.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/gemv_masked.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/hadamard.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/hadamard.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/gather.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/gather.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/gather_axis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/gather_axis.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/gather_front.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/gather_front.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/indexing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/indexing.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/masked_scatter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/masked_scatter.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/scatter.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/scatter.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/indexing/scatter_axis.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/indexing/scatter_axis.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/layer_norm.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/layer_norm.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/logsumexp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/logsumexp.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/logsumexp.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/logsumexp.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/quantized.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/quantized.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/quantized.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/quantized.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/quantized_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/quantized_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/quantized_nax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/quantized_nax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/quantized_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/quantized_utils.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/random.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/random.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduce.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduce.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduce.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduce_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduce_utils.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduction/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduction/ops.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduction/reduce_all.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduction/reduce_all.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduction/reduce_col.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduction/reduce_col.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduction/reduce_init.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduction/reduce_init.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/reduction/reduce_row.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/reduction/reduce_row.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/rms_norm.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/rms_norm.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/rope.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/rope.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/scaled_dot_product_attention.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/scaled_dot_product_attention.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/scan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/scan.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/scan.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/scan.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/sdpa_vector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/sdpa_vector.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/softmax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/softmax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/softmax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/softmax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/sort.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/sort.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/sort.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/sort.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/attn.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/loader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/loader.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/mma.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/mma.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/params.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/params.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/attn/transforms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/attn/transforms.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/conv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/conv.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/kernels/steel_conv_general.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/loader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/loader.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/conv/params.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/conv/params.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/defines.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/defines.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/gemm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/gemm.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/gemm_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/gemm_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/loader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/loader.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/mma.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/mma.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/nax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/nax.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/params.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/params.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/gemm/transforms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/gemm/transforms.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/utils.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/utils/integral_constant.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/utils/integral_constant.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/steel/utils/type_traits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/steel/utils/type_traits.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/ternary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/ternary.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/ternary.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/ternary.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/ternary_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/ternary_ops.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/unary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/unary.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/unary.metal: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/unary.metal -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/unary_ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/unary_ops.h -------------------------------------------------------------------------------- /mlx/backend/metal/kernels/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/kernels/utils.h -------------------------------------------------------------------------------- /mlx/backend/metal/logsumexp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/logsumexp.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/make_compiled_preamble.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/make_compiled_preamble.sh -------------------------------------------------------------------------------- /mlx/backend/metal/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/matmul.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/matmul.h -------------------------------------------------------------------------------- /mlx/backend/metal/metal.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/metal.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/metal.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/metal.h -------------------------------------------------------------------------------- /mlx/backend/metal/no_metal.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/no_metal.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/nojit_kernels.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/nojit_kernels.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/normalization.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/normalization.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/primitives.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/quantized.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/quantized.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/reduce.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/reduce.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/reduce.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/reduce.h -------------------------------------------------------------------------------- /mlx/backend/metal/resident.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/resident.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/resident.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/resident.h -------------------------------------------------------------------------------- /mlx/backend/metal/rope.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/rope.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/scaled_dot_product_attention.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/scaled_dot_product_attention.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/scan.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/scan.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/scan.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/scan.h -------------------------------------------------------------------------------- /mlx/backend/metal/slicing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/slicing.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/softmax.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/softmax.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/sort.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/sort.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/ternary.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/ternary.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/ternary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/ternary.h -------------------------------------------------------------------------------- /mlx/backend/metal/unary.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/unary.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/unary.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/unary.h -------------------------------------------------------------------------------- /mlx/backend/metal/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/utils.cpp -------------------------------------------------------------------------------- /mlx/backend/metal/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/metal/utils.h -------------------------------------------------------------------------------- /mlx/backend/no_cpu/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_cpu/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/no_cpu/available.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_cpu/available.cpp -------------------------------------------------------------------------------- /mlx/backend/no_cpu/compiled.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_cpu/compiled.cpp -------------------------------------------------------------------------------- /mlx/backend/no_cpu/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_cpu/primitives.cpp -------------------------------------------------------------------------------- /mlx/backend/no_gpu/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/backend/no_gpu/allocator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/allocator.cpp -------------------------------------------------------------------------------- /mlx/backend/no_gpu/apple_memory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/apple_memory.h -------------------------------------------------------------------------------- /mlx/backend/no_gpu/eval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/eval.cpp -------------------------------------------------------------------------------- /mlx/backend/no_gpu/event.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/event.cpp -------------------------------------------------------------------------------- /mlx/backend/no_gpu/fence.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/fence.cpp -------------------------------------------------------------------------------- /mlx/backend/no_gpu/linux_memory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/linux_memory.h -------------------------------------------------------------------------------- /mlx/backend/no_gpu/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/backend/no_gpu/primitives.cpp -------------------------------------------------------------------------------- /mlx/compile.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/compile.cpp -------------------------------------------------------------------------------- /mlx/compile.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/compile.h -------------------------------------------------------------------------------- /mlx/compile_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/compile_impl.h -------------------------------------------------------------------------------- /mlx/device.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/device.cpp -------------------------------------------------------------------------------- /mlx/device.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/device.h -------------------------------------------------------------------------------- /mlx/distributed/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/distributed/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/distributed.cpp -------------------------------------------------------------------------------- /mlx/distributed/distributed.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/distributed.h -------------------------------------------------------------------------------- /mlx/distributed/distributed_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/distributed_impl.h -------------------------------------------------------------------------------- /mlx/distributed/mpi/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/mpi/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/distributed/mpi/mpi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/mpi/mpi.cpp -------------------------------------------------------------------------------- /mlx/distributed/mpi/mpi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/mpi/mpi.h -------------------------------------------------------------------------------- /mlx/distributed/mpi/mpi_declarations.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/mpi/mpi_declarations.h -------------------------------------------------------------------------------- /mlx/distributed/mpi/no_mpi.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/mpi/no_mpi.cpp -------------------------------------------------------------------------------- /mlx/distributed/nccl/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/distributed/nccl/nccl.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/nccl.cpp -------------------------------------------------------------------------------- /mlx/distributed/nccl/nccl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/nccl.h -------------------------------------------------------------------------------- /mlx/distributed/nccl/nccl_stub/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/nccl_stub/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/distributed/nccl/nccl_stub/nccl_stubs.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/nccl_stub/nccl_stubs.cpp -------------------------------------------------------------------------------- /mlx/distributed/nccl/no_nccl.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/nccl/no_nccl.cpp -------------------------------------------------------------------------------- /mlx/distributed/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ops.cpp -------------------------------------------------------------------------------- /mlx/distributed/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ops.h -------------------------------------------------------------------------------- /mlx/distributed/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/primitives.cpp -------------------------------------------------------------------------------- /mlx/distributed/primitives.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/primitives.h -------------------------------------------------------------------------------- /mlx/distributed/ring/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ring/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/distributed/ring/no_ring.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ring/no_ring.cpp -------------------------------------------------------------------------------- /mlx/distributed/ring/ring.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ring/ring.cpp -------------------------------------------------------------------------------- /mlx/distributed/ring/ring.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/distributed/ring/ring.h -------------------------------------------------------------------------------- /mlx/dtype.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/dtype.cpp -------------------------------------------------------------------------------- /mlx/dtype.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/dtype.h -------------------------------------------------------------------------------- /mlx/dtype_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/dtype_utils.cpp -------------------------------------------------------------------------------- /mlx/dtype_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/dtype_utils.h -------------------------------------------------------------------------------- /mlx/einsum.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/einsum.cpp -------------------------------------------------------------------------------- /mlx/einsum.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/einsum.h -------------------------------------------------------------------------------- /mlx/event.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/event.h -------------------------------------------------------------------------------- /mlx/export.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/export.cpp -------------------------------------------------------------------------------- /mlx/export.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/export.h -------------------------------------------------------------------------------- /mlx/export_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/export_impl.h -------------------------------------------------------------------------------- /mlx/fast.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fast.cpp -------------------------------------------------------------------------------- /mlx/fast.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fast.h -------------------------------------------------------------------------------- /mlx/fast_primitives.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fast_primitives.h -------------------------------------------------------------------------------- /mlx/fence.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fence.h -------------------------------------------------------------------------------- /mlx/fft.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fft.cpp -------------------------------------------------------------------------------- /mlx/fft.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/fft.h -------------------------------------------------------------------------------- /mlx/graph_utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/graph_utils.cpp -------------------------------------------------------------------------------- /mlx/graph_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/graph_utils.h -------------------------------------------------------------------------------- /mlx/io.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io.h -------------------------------------------------------------------------------- /mlx/io/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/CMakeLists.txt -------------------------------------------------------------------------------- /mlx/io/gguf.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/gguf.cpp -------------------------------------------------------------------------------- /mlx/io/gguf.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/gguf.h -------------------------------------------------------------------------------- /mlx/io/gguf_quants.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/gguf_quants.cpp -------------------------------------------------------------------------------- /mlx/io/load.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/load.cpp -------------------------------------------------------------------------------- /mlx/io/load.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/load.h -------------------------------------------------------------------------------- /mlx/io/no_gguf.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/no_gguf.cpp -------------------------------------------------------------------------------- /mlx/io/no_safetensors.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/no_safetensors.cpp -------------------------------------------------------------------------------- /mlx/io/safetensors.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/io/safetensors.cpp -------------------------------------------------------------------------------- /mlx/linalg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/linalg.cpp -------------------------------------------------------------------------------- /mlx/linalg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/linalg.h -------------------------------------------------------------------------------- /mlx/memory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/memory.h -------------------------------------------------------------------------------- /mlx/mlx.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/mlx.h -------------------------------------------------------------------------------- /mlx/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/ops.cpp -------------------------------------------------------------------------------- /mlx/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/ops.h -------------------------------------------------------------------------------- /mlx/primitives.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/primitives.cpp -------------------------------------------------------------------------------- /mlx/primitives.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/primitives.h -------------------------------------------------------------------------------- /mlx/random.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/random.cpp -------------------------------------------------------------------------------- /mlx/random.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/random.h -------------------------------------------------------------------------------- /mlx/scheduler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/scheduler.cpp -------------------------------------------------------------------------------- /mlx/scheduler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/scheduler.h -------------------------------------------------------------------------------- /mlx/small_vector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/small_vector.h -------------------------------------------------------------------------------- /mlx/stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/stream.h -------------------------------------------------------------------------------- /mlx/threadpool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/threadpool.h -------------------------------------------------------------------------------- /mlx/transforms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/transforms.cpp -------------------------------------------------------------------------------- /mlx/transforms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/transforms.h -------------------------------------------------------------------------------- /mlx/transforms_impl.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/transforms_impl.h -------------------------------------------------------------------------------- /mlx/types/bf16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/types/bf16.h -------------------------------------------------------------------------------- /mlx/types/complex.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/types/complex.h -------------------------------------------------------------------------------- /mlx/types/fp16.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/types/fp16.h -------------------------------------------------------------------------------- /mlx/types/half_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/types/half_types.h -------------------------------------------------------------------------------- /mlx/types/limits.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/types/limits.h -------------------------------------------------------------------------------- /mlx/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/utils.cpp -------------------------------------------------------------------------------- /mlx/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/utils.h -------------------------------------------------------------------------------- /mlx/version.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/version.cpp -------------------------------------------------------------------------------- /mlx/version.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/mlx/version.h -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/pyproject.toml -------------------------------------------------------------------------------- /python/mlx/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/__main__.py -------------------------------------------------------------------------------- /python/mlx/_reprlib_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/_reprlib_fix.py -------------------------------------------------------------------------------- /python/mlx/_stub_patterns.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/_stub_patterns.txt -------------------------------------------------------------------------------- /python/mlx/distributed_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/distributed_run.py -------------------------------------------------------------------------------- /python/mlx/extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/extension.py -------------------------------------------------------------------------------- /python/mlx/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/__init__.py -------------------------------------------------------------------------------- /python/mlx/nn/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/init.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/__init__.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/activations.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/base.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/containers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/containers.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/convolution.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/convolution_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/convolution_transpose.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/distributed.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/dropout.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/embedding.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/linear.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/normalization.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/pooling.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/positional_encoding.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/quantized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/quantized.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/recurrent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/recurrent.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/transformer.py -------------------------------------------------------------------------------- /python/mlx/nn/layers/upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/layers/upsample.py -------------------------------------------------------------------------------- /python/mlx/nn/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/losses.py -------------------------------------------------------------------------------- /python/mlx/nn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/nn/utils.py -------------------------------------------------------------------------------- /python/mlx/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/optimizers/__init__.py -------------------------------------------------------------------------------- /python/mlx/optimizers/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/optimizers/optimizers.py -------------------------------------------------------------------------------- /python/mlx/optimizers/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/optimizers/schedulers.py -------------------------------------------------------------------------------- /python/mlx/py.typed: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /python/mlx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/mlx/utils.py -------------------------------------------------------------------------------- /python/scripts/repair_cuda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/scripts/repair_cuda.sh -------------------------------------------------------------------------------- /python/scripts/repair_linux.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/scripts/repair_linux.sh -------------------------------------------------------------------------------- /python/scripts/repair_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/scripts/repair_record.py -------------------------------------------------------------------------------- /python/src/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/CMakeLists.txt -------------------------------------------------------------------------------- /python/src/array.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/array.cpp -------------------------------------------------------------------------------- /python/src/buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/buffer.h -------------------------------------------------------------------------------- /python/src/constants.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/constants.cpp -------------------------------------------------------------------------------- /python/src/convert.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/convert.cpp -------------------------------------------------------------------------------- /python/src/convert.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/convert.h -------------------------------------------------------------------------------- /python/src/cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/cuda.cpp -------------------------------------------------------------------------------- /python/src/device.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/device.cpp -------------------------------------------------------------------------------- /python/src/distributed.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/distributed.cpp -------------------------------------------------------------------------------- /python/src/export.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/export.cpp -------------------------------------------------------------------------------- /python/src/fast.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/fast.cpp -------------------------------------------------------------------------------- /python/src/fft.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/fft.cpp -------------------------------------------------------------------------------- /python/src/indexing.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/indexing.cpp -------------------------------------------------------------------------------- /python/src/indexing.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/indexing.h -------------------------------------------------------------------------------- /python/src/linalg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/linalg.cpp -------------------------------------------------------------------------------- /python/src/load.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/load.cpp -------------------------------------------------------------------------------- /python/src/load.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/load.h -------------------------------------------------------------------------------- /python/src/memory.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/memory.cpp -------------------------------------------------------------------------------- /python/src/metal.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/metal.cpp -------------------------------------------------------------------------------- /python/src/mlx.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/mlx.cpp -------------------------------------------------------------------------------- /python/src/mlx_func.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/mlx_func.cpp -------------------------------------------------------------------------------- /python/src/mlx_func.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/mlx_func.h -------------------------------------------------------------------------------- /python/src/ops.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/ops.cpp -------------------------------------------------------------------------------- /python/src/random.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/random.cpp -------------------------------------------------------------------------------- /python/src/small_vector.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/small_vector.h -------------------------------------------------------------------------------- /python/src/stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/stream.cpp -------------------------------------------------------------------------------- /python/src/transforms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/transforms.cpp -------------------------------------------------------------------------------- /python/src/trees.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/trees.cpp -------------------------------------------------------------------------------- /python/src/trees.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/trees.h -------------------------------------------------------------------------------- /python/src/utils.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/utils.cpp -------------------------------------------------------------------------------- /python/src/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/src/utils.h -------------------------------------------------------------------------------- /python/tests/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/__main__.py -------------------------------------------------------------------------------- /python/tests/cuda_skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/cuda_skip.py -------------------------------------------------------------------------------- /python/tests/mlx_distributed_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/mlx_distributed_tests.py -------------------------------------------------------------------------------- /python/tests/mlx_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/mlx_tests.py -------------------------------------------------------------------------------- /python/tests/mpi_test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/mpi_test_distributed.py -------------------------------------------------------------------------------- /python/tests/nccl_test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/nccl_test_distributed.py -------------------------------------------------------------------------------- /python/tests/ring_test_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/ring_test_distributed.py -------------------------------------------------------------------------------- /python/tests/test_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_array.py -------------------------------------------------------------------------------- /python/tests/test_autograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_autograd.py -------------------------------------------------------------------------------- /python/tests/test_bf16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_bf16.py -------------------------------------------------------------------------------- /python/tests/test_blas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_blas.py -------------------------------------------------------------------------------- /python/tests/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_compile.py -------------------------------------------------------------------------------- /python/tests/test_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_constants.py -------------------------------------------------------------------------------- /python/tests/test_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_conv.py -------------------------------------------------------------------------------- /python/tests/test_conv_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_conv_transpose.py -------------------------------------------------------------------------------- /python/tests/test_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_device.py -------------------------------------------------------------------------------- /python/tests/test_double.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_double.py -------------------------------------------------------------------------------- /python/tests/test_einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_einsum.py -------------------------------------------------------------------------------- /python/tests/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_eval.py -------------------------------------------------------------------------------- /python/tests/test_export_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_export_import.py -------------------------------------------------------------------------------- /python/tests/test_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_fast.py -------------------------------------------------------------------------------- /python/tests/test_fast_sdpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_fast_sdpa.py -------------------------------------------------------------------------------- /python/tests/test_fft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_fft.py -------------------------------------------------------------------------------- /python/tests/test_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_graph.py -------------------------------------------------------------------------------- /python/tests/test_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_init.py -------------------------------------------------------------------------------- /python/tests/test_linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_linalg.py -------------------------------------------------------------------------------- /python/tests/test_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_load.py -------------------------------------------------------------------------------- /python/tests/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_losses.py -------------------------------------------------------------------------------- /python/tests/test_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_memory.py -------------------------------------------------------------------------------- /python/tests/test_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_nn.py -------------------------------------------------------------------------------- /python/tests/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_ops.py -------------------------------------------------------------------------------- /python/tests/test_optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_optimizers.py -------------------------------------------------------------------------------- /python/tests/test_quantized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_quantized.py -------------------------------------------------------------------------------- /python/tests/test_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_random.py -------------------------------------------------------------------------------- /python/tests/test_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_reduce.py -------------------------------------------------------------------------------- /python/tests/test_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_tree.py -------------------------------------------------------------------------------- /python/tests/test_upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_upsample.py -------------------------------------------------------------------------------- /python/tests/test_vmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/python/tests/test_vmap.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/setup.py -------------------------------------------------------------------------------- /tests/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/CMakeLists.txt -------------------------------------------------------------------------------- /tests/allocator_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/allocator_tests.cpp -------------------------------------------------------------------------------- /tests/arg_reduce_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/arg_reduce_tests.cpp -------------------------------------------------------------------------------- /tests/array_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/array_tests.cpp -------------------------------------------------------------------------------- /tests/autograd_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/autograd_tests.cpp -------------------------------------------------------------------------------- /tests/blas_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/blas_tests.cpp -------------------------------------------------------------------------------- /tests/compile_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/compile_tests.cpp -------------------------------------------------------------------------------- /tests/creations_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/creations_tests.cpp -------------------------------------------------------------------------------- /tests/custom_vjp_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/custom_vjp_tests.cpp -------------------------------------------------------------------------------- /tests/device_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/device_tests.cpp -------------------------------------------------------------------------------- /tests/einsum_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/einsum_tests.cpp -------------------------------------------------------------------------------- /tests/eval_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/eval_tests.cpp -------------------------------------------------------------------------------- /tests/export_import_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/export_import_tests.cpp -------------------------------------------------------------------------------- /tests/fft_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/fft_tests.cpp -------------------------------------------------------------------------------- /tests/gpu_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/gpu_tests.cpp -------------------------------------------------------------------------------- /tests/linalg_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/linalg_tests.cpp -------------------------------------------------------------------------------- /tests/load_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/load_tests.cpp -------------------------------------------------------------------------------- /tests/ops_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/ops_tests.cpp -------------------------------------------------------------------------------- /tests/random_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/random_tests.cpp -------------------------------------------------------------------------------- /tests/scheduler_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/scheduler_tests.cpp -------------------------------------------------------------------------------- /tests/tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/tests.cpp -------------------------------------------------------------------------------- /tests/utils_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/utils_tests.cpp -------------------------------------------------------------------------------- /tests/vmap_tests.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-explore/mlx/HEAD/tests/vmap_tests.cpp --------------------------------------------------------------------------------