Here is a list of all files with brief descriptions:

[detail level 12345678]

▼ mlx
▼ 3rdparty
pocketfft.h
▼ backend
▼ accelerate
utils.h
▼ common
arange.h
binary.h
binary_two.h
compiled.h
compiled_preamble.h
copy.h
lapack_helper.h
ops.h
reduce.h
slicing.h
ternary.h
threefry.h
unary.h
utils.h
▼ metal
► jit
arange.h
binary.h
binary_two.h
copy.h
fft.h
includes.h
indexing.h
reduce.h
scan.h
softmax.h
sort.h
steel_conv.h
steel_gemm.h
ternary.h
unary.h
► kernels
► fft
radix.h
readwrite.h
► reduction
ops.h
reduce_all.h
reduce_col.h
reduce_init.h
reduce_row.h
► steel
► conv
► kernels
steel_conv.h
steel_conv_general.h
► loaders
loader_channel_l.h
loader_channel_n.h
loader_general.h
conv.h
loader.h
params.h
► gemm
► kernels
steel_gemm_fused.h
steel_gemm_masked.h
steel_gemm_splitk.h
gemm.h
loader.h
mma.h
params.h
transforms.h
defines.h
utils.h
arange.h
atomic.h
bf16.h
bf16_math.h
binary.h
binary_ops.h
binary_two.h
complex.h
copy.h
defines.h
erf.h
expm1f.h
fft.h
gather.h
indexing.h
reduce.h
reduce_utils.h
scaled_dot_product_attention_params.h
scan.h
scatter.h
softmax.h
sort.h
ternary.h
ternary_ops.h
unary.h
unary_ops.h
utils.h
► mps
gemm.h
allocator.h
binary.h
copy.h
device.h
kernels.h
matmul.h
metal.h
metal_impl.h
reduce.h
slicing.h
ternary.h
unary.h
utils.h
▼ distributed
distributed.h
ops.h
primitives.h
▼ io
gguf.h
load.h
▼ types
bf16.h
complex.h
fp16.h
half_types.h
allocator.h
array.h
compile.h
compile_impl.h
device.h
dtype.h
event.h
fast.h
fast_primitives.h
fft.h
graph_utils.h
io.h
linalg.h
mlx.h
ops.h
primitives.h
random.h
scheduler.h
stream.h
transforms.h
transforms_impl.h
utils.h