LICENSE
MANIFEST.in
README.md
pyproject.toml
mhc_mlx/__init__.py
mhc_mlx/diagnostics.py
mhc_mlx/layer.py
mhc_mlx/metal.py
mhc_mlx/reference.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/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_correctness.py
tests/test_packaging.py
tests/test_stress.py