LICENSE
NOTICE
README.md
pyproject.toml
tests/test_benchmark_integrity.py
tests/test_codegen.py
tests/test_direct_commits.py
tests/test_epilogues_gpu.py
tests/test_frontend.py
tests/test_gemv.py
tests/test_gpu.py
tests/test_h100_evidence.py
tests/test_h100_heldout_script.py
tests/test_h100_tuning.py
tests/test_h100_validation.py
tests/test_inplace_training.py
tests/test_ir_reference.py
tests/test_ir_validation.py
tests/test_optimizer_safety.py
tests/test_side_epilogues.py
tests/test_sm90.py
tests/test_sm90_probe.py
tests/test_summarize_h100.py
tests/test_train_loop.py
tests/test_training_api.py
tests/test_transpose_pass.py
tests/test_wmma.py
trainkernel/__init__.py
trainkernel/codegen.py
trainkernel/frontend.py
trainkernel/gemv.py
trainkernel/ir.py
trainkernel/passes.py
trainkernel/py.typed
trainkernel/reference.py
trainkernel/runtime.py
trainkernel/sm90.py
trainkernel/wmma.py
trainkernel.egg-info/PKG-INFO
trainkernel.egg-info/SOURCES.txt
trainkernel.egg-info/dependency_links.txt
trainkernel.egg-info/requires.txt
trainkernel.egg-info/top_level.txt