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
ternary.h
threefry.h
unary.h
utils.h
▼ metal
▼ kernels
▼ reduction
ops.h
reduce_inst.h
utils.h
▼ steel
▼ conv
▼ loaders
loader_channel_l.h
loader_channel_n.h
loader_general.h
conv.h
loader.h
params.h
▼ gemm
gemm.h
loader.h
mma.h
params.h
transforms.h
utils.h
atomic.h
bf16.h
bf16_math.h
binary.h
compiled_preamble.h
complex.h
defines.h
erf.h
expm1f.h
indexing.h
scaled_dot_product_attention_params.h
ternary.h
unary.h
utils.h
▼ mps
gemm.h
allocator.h
compiled_preamble.h
copy.h
device.h
matmul.h
metal.h
metal_impl.h
reduce.h
utils.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