LICENSE
MANIFEST.in
README.md
pyproject.toml
mhc_mlx/__init__.py
mhc_mlx/benchmark.py
mhc_mlx/diagnostics.py
mhc_mlx/kernels_embedded.py
mhc_mlx/layer.py
mhc_mlx/metal.py
mhc_mlx/patching.py
mhc_mlx/reference.py
mhc_mlx/utils.py
mhc_mlx.egg-info/PKG-INFO
mhc_mlx.egg-info/SOURCES.txt
mhc_mlx.egg-info/dependency_links.txt
mhc_mlx.egg-info/entry_points.txt
mhc_mlx.egg-info/requires.txt
mhc_mlx.egg-info/top_level.txt
mhc_mlx/kernels/mhc_backward_dH_post.metal
mhc_mlx/kernels/mhc_backward_dH_pre.metal
mhc_mlx/kernels/mhc_backward_dH_pre_post.metal
mhc_mlx/kernels/mhc_backward_dM.metal
mhc_mlx/kernels/mhc_backward_d_rms_weight.metal
mhc_mlx/kernels/mhc_backward_dx.metal
mhc_mlx/kernels/mhc_backward_dx_col.metal
mhc_mlx/kernels/mhc_backward_fused_dx.metal
mhc_mlx/kernels/mhc_backward_grads_fused.metal
mhc_mlx/kernels/mhc_backward_prep.metal
mhc_mlx/kernels/mhc_backward_prep_tile.metal
mhc_mlx/kernels/mhc_backward_rms_reduce.metal
mhc_mlx/kernels/mhc_forward_agg.metal
mhc_mlx/kernels/mhc_forward_agg_bf16.metal
mhc_mlx/kernels/mhc_forward_rms_reduce.metal
mhc_mlx/kernels/mhc_fused.metal
mhc_mlx/kernels/residual_add_agg.metal
mhc_mlx/kernels/sinkhorn_knopp.metal
mhc_mlx/kernels/sinkhorn_knopp_backward.metal
mhc_mlx/kernels/stream_mix_add.metal
mhc_mlx/kernels/stream_mix_add_rms.metal
mhc_mlx/kernels/stream_mix_add_rms_bf16.metal
mhc_mlx/kernels/stream_mix_add_rms_col.metal
mhc_mlx/kernels/stream_mix_add_rms_col_bf16.metal
mhc_mlx/kernels/stream_mix_add_rms_fp16.metal
mhc_mlx/kernels/stream_mix_add_rms_tile.metal
mhc_mlx/kernels/stream_mix_add_rms_tile2d_bf16.metal
mhc_mlx/kernels/stream_mix_add_rms_tile2d_fp16.metal
mhc_mlx/kernels/stream_mix_add_rms_tile_bf16.metal
mhc_mlx/kernels/stream_mix_add_rms_tile_f32.metal
mhc_mlx/kernels/stream_mix_add_rms_tile_fp16.metal
mhc_mlx/kernels/stream_mix_backward_dx.metal
mlx_mhc/__init__.py
tests/test_concurrency.py
tests/test_correctness.py
tests/test_determinism.py
tests/test_folding.py
tests/test_packaging.py
tests/test_patching.py
tests/test_quantized.py
tests/test_rewire.py
tests/test_stability.py
tests/test_stress.py
tests/test_vision.py