.gitignore
.pre-commit-config.yaml
CHANGELOG.md
CLAUDE.md
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
README.md
mkdocs.yml
pyproject.toml
uv.lock
.github/workflows/ci.yml
.github/workflows/notify-umbrella.yml
.github/workflows/publish.yml
benchmarks/__init__.py
benchmarks/bench_einsum.py
docs/architecture.md
docs/aws_setup.md
docs/benchmarks.md
docs/contraction_patterns.md
docs/developing_kernels.md
docs/index.md
docs/installation.md
docs/quickstart.md
docs/tutorial_df_mp2.md
docs/api/decompose.md
docs/api/einsum.md
docs/api/index.md
docs/api/nki.md
docs/api/plan.md
docs/api/quantum.md
docs/design/matmul_dispatch_findings.md
docs/design/matmul_kernel_profile.md
examples/df_mp2_einsum.py
infra/terraform/README.md
infra/terraform/main.tf
scripts/autotune_dispatch.py
scripts/bench_native_einsum.py
scripts/bench_native_kernels.py
scripts/bench_to_md.py
scripts/run_benchmarks.sh
scripts/run_neuron_profile.sh
scripts/run_neuron_tests.sh
scripts/run_reduce_parallel_multirank.py
scripts/run_simulator_tests.sh
tests/conftest.py
tests/test_decompose.py
tests/test_einsum.py
tests/test_host_output.py
tests/test_native_dispatch.py
tests/test_nki.py
tests/test_nki_compile.py
tests/test_nki_kernels.py
tests/test_nki_native.py
tests/test_nki_sim.py
tests/test_parallel.py
tests/test_plan.py
tests/test_quantum.py
tests/test_sr_runtime.py
trntensor/__init__.py
trntensor/decompose.py
trntensor/einsum.py
trntensor/parallel.py
trntensor/plan.py
trntensor/py.typed
trntensor/quantum.py
trntensor.egg-info/PKG-INFO
trntensor.egg-info/SOURCES.txt
trntensor.egg-info/dependency_links.txt
trntensor.egg-info/requires.txt
trntensor.egg-info/top_level.txt
trntensor/nki/__init__.py
trntensor/nki/_kernels.py
trntensor/nki/_native.py
trntensor/nki/collectives.py
trntensor/nki/dispatch.py