.editorconfig
.gitignore
LICENSE
README.md
pyproject.toml
tests/__init__.py
tests/conftest.py
tests/pytest_benchmark_plugin.py
tests/pytest_random_plugin.py
tests/engram/test_engram_fused_weight.py
tests/engram/test_engram_gate_bwd.py
tests/engram/test_engram_gate_fwd.py
tests/engram/test_engram_grad_w_reduce.py
tests/engram/test_engram_hash.py
tests/mhc/test_expand.py
tests/mhc/test_head_compute_mix.py
tests/mhc/test_multilayer_recompute.py
tests/mhc/test_norm_fn.py
tests/mhc/test_post.py
tests/mhc/test_pre_apply_mix.py
tests/mhc/test_pre_big_fuse.py
tests/mhc/test_pre_split_mixes.py
tests/mhc/test_sinkhorn.py
tests/moe/test_aux_fi.py
tests/moe/test_expand_to_fused.py
tests/moe/test_get_fused_mapping.py
tests/moe/test_group_count.py
tests/moe/test_inplace_unique_group_indices.py
tests/moe/test_mask_indices_by_tp.py
tests/moe/test_normalize_weight.py
tests/moe/test_reduce_fused.py
tests/moe/test_top2_sum_gate.py
tests/moe/test_topk_gate.py
tests/moe/test_topk_sum_and_topk_idx.py
tests/quant/test_cast_back.py
tests/quant/test_cast_back_e5m6.py
tests/quant/test_per_block_cast.py
tests/quant/test_per_block_cast_lossless.py
tests/quant/test_per_channel_cast.py
tests/quant/test_per_channel_cast_and_transpose.py
tests/quant/test_per_channel_cast_fused.py
tests/quant/test_per_token_cast.py
tests/quant/test_per_token_cast_to_e5m6.py
tests/quant/test_swiglu_backward_and_per_token_cast.py
tests/quant/test_swiglu_forward_and_per_channel_cast_and_transpose.py
tests/quant/test_swiglu_forward_and_per_token_cast.py
tests/transpose/test_transpose.py
tile_kernels/__init__.py
tile_kernels/_version.py
tile_kernels/config.py
tile_kernels/utils.py
tile_kernels.egg-info/PKG-INFO
tile_kernels.egg-info/SOURCES.txt
tile_kernels.egg-info/dependency_links.txt
tile_kernels.egg-info/requires.txt
tile_kernels.egg-info/top_level.txt
tile_kernels/engram/__init__.py
tile_kernels/engram/engram_fused_weight_kernel.py
tile_kernels/engram/engram_gate_kernel.py
tile_kernels/engram/engram_grad_w_reduce_kernel.py
tile_kernels/engram/engram_hash_kernel.py
tile_kernels/mhc/__init__.py
tile_kernels/mhc/expand_kernel.py
tile_kernels/mhc/head_compute_mix_kernel.py
tile_kernels/mhc/multilayer_recompute_kernel.py
tile_kernels/mhc/norm_fn_kernel.py
tile_kernels/mhc/post_kernel.py
tile_kernels/mhc/pre_apply_mix_kernel.py
tile_kernels/mhc/pre_big_fuse_kernel.py
tile_kernels/mhc/pre_split_mixes_kernel.py
tile_kernels/mhc/sinkhorn_kernel.py
tile_kernels/modeling/__init__.py
tile_kernels/modeling/engram/__init__.py
tile_kernels/modeling/engram/engram_gate.py
tile_kernels/modeling/mhc/__init__.py
tile_kernels/modeling/mhc/functional.py
tile_kernels/modeling/mhc/ops/__init__.py
tile_kernels/modeling/mhc/ops/expand.py
tile_kernels/modeling/mhc/ops/head_compute_mix.py
tile_kernels/modeling/mhc/ops/multilayer_recompute.py
tile_kernels/modeling/mhc/ops/norm_fn.py
tile_kernels/modeling/mhc/ops/post.py
tile_kernels/modeling/mhc/ops/pre_apply_mix.py
tile_kernels/modeling/mhc/ops/pre_big_fuse.py
tile_kernels/modeling/mhc/ops/pre_split_mixes.py
tile_kernels/modeling/mhc/ops/sinkhorn.py
tile_kernels/moe/__init__.py
tile_kernels/moe/aux_fi_kernel.py
tile_kernels/moe/common.py
tile_kernels/moe/expand_to_fused_kernel.py
tile_kernels/moe/get_fused_mapping_kernel.py
tile_kernels/moe/group_count_kernel.py
tile_kernels/moe/inplace_unique_group_indices_kernel.py
tile_kernels/moe/mask_indices_by_tp_kernel.py
tile_kernels/moe/normalize_weight_kernel.py
tile_kernels/moe/reduce_fused_kernel.py
tile_kernels/moe/scoring.py
tile_kernels/moe/top2_sum_gate_kernel.py
tile_kernels/moe/topk_gate_kernel.py
tile_kernels/moe/topk_sum_and_topk_group_idx_kernel.py
tile_kernels/quant/__init__.py
tile_kernels/quant/cast_back_e5m6_kernel.py
tile_kernels/quant/cast_back_kernel.py
tile_kernels/quant/common.py
tile_kernels/quant/per_block_cast_kernel.py
tile_kernels/quant/per_block_cast_lossless_kernel.py
tile_kernels/quant/per_channel_cast_and_transpose_kernel.py
tile_kernels/quant/per_channel_cast_fused_kernel.py
tile_kernels/quant/per_channel_cast_kernel.py
tile_kernels/quant/per_token_cast_kernel.py
tile_kernels/quant/per_token_cast_to_e5m6_kernel.py
tile_kernels/quant/swiglu_backward_and_per_token_cast_kernel.py
tile_kernels/quant/swiglu_forward_and_per_channel_cast_and_transpose_kernel.py
tile_kernels/quant/swiglu_forward_and_per_token_cast_kernel.py
tile_kernels/quant/types.py
tile_kernels/testing/__init__.py
tile_kernels/testing/bench.py
tile_kernels/testing/generator.py
tile_kernels/testing/numeric.py
tile_kernels/testing/quant.py
tile_kernels/torch/__init__.py
tile_kernels/torch/cast.py
tile_kernels/torch/cast_e5m6.py
tile_kernels/torch/engram.py
tile_kernels/torch/expand_to_fused.py
tile_kernels/torch/mhc.py
tile_kernels/torch/moe.py
tile_kernels/torch/per_channel_cast_fused.py
tile_kernels/torch/reduce_fused.py
tile_kernels/torch/swiglu.py
tile_kernels/torch/topk.py
tile_kernels/transpose/__init__.py
tile_kernels/transpose/batched_transpose_kernel.py