.dockerignore
.gitignore
LICENSE.md
MANIFEST.in
README.md
compile_layers.sh
docs-requirements.txt
engine.py
mkdocs.yml
pyproject.toml
requirements.txt
server.py
.github/scripts/release.py
.github/workflows/gh-pages.yml
.github/workflows/publish.yml
.github/workflows/release.yml
assets/images/AryaXAI-light.jpg
assets/images/AryaXAI.png
assets/images/arya-fav.png
assets/images/arya-logo-mini.png
assets/images/dlb_logo.png
assets/images/lexsi_logo.svg
assets/images/speedup_comparison.png
assets/images/throughput_comparison.png
assets/images/total_dlb_time_comparison.png
dl_backtrace/__init__.py
dl_backtrace/version.py
dl_backtrace.egg-info/PKG-INFO
dl_backtrace.egg-info/SOURCES.txt
dl_backtrace.egg-info/dependency_links.txt
dl_backtrace.egg-info/entry_points.txt
dl_backtrace.egg-info/requires.txt
dl_backtrace.egg-info/top_level.txt
dl_backtrace/cli/__init__.py
dl_backtrace/cli/engine.py
dl_backtrace/cli/server.py
dl_backtrace/moe_pytorch_backtrace/__init__.py
dl_backtrace/moe_pytorch_backtrace/backtrace/__init__.py
dl_backtrace/moe_pytorch_backtrace/backtrace/backtrace.py
dl_backtrace/moe_pytorch_backtrace/backtrace/config.py
dl_backtrace/moe_pytorch_backtrace/backtrace/supported_models.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/gpt_oss.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/helper.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/jetmoe.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/model_utils.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/moe_auto_sampler.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/olmoe.py
dl_backtrace/moe_pytorch_backtrace/backtrace/core/qwen3_moe.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/__init__.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/default_v2.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/__init__.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/GPT_oss/original_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/GPT_oss/pytorch_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/GPT_oss/refactored_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/GPT_oss/helper_cuda/lm_head.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/JetMoE/original_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/JetMoE/pytorch_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/JetMoE/refactored_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/OLMoE/original_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/OLMoE/pytorch_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/OLMoE/refactored_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/Qwen_MoE/original_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/Qwen_MoE/pytorch_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/Qwen_MoE/refactored_version.py
dl_backtrace/moe_pytorch_backtrace/backtrace/utils/cuda_utils/Qwen_MoE/helper_cuda/lm_head.py
dl_backtrace/pytorch_backtrace/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/activation.py
dl_backtrace/pytorch_backtrace/dlbacktrace/aten_operations.py
dl_backtrace/pytorch_backtrace/dlbacktrace/dlbacktrace.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/compiled_propagation.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/config.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/dlb_auto_sampler.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/execution_engine_noncache.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/graph_builder.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/io_utils.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/relevance_propagation.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/relevance_saver.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/reproducibility.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/token_relevance_visuals.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/trace_utils.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/visualization.py
dl_backtrace/pytorch_backtrace/dlbacktrace/core/visualization_module_aware.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/default_v2.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/calculate_gavgpool2d_kernel.cu
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/calculate_gavgpool2d_op.cpp
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/setup.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/wt_gavgpool_ops.cpython-310-x86_64-linux-gnu.so
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/wt_gavgpool_ops.cpython-312-x86_64-linux-gnu.so
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/include/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AdaptiveAvgPool2D/cuda_version/include/utils.h
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AveragePool2D/cuda_padding.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AveragePool2D/cuda_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AveragePool2D/cuda_wt_avg_unit.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/AveragePool2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/cuda_padding.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/cuda_wt_conv_unit.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/launch_cuda_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Conv2D/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/ConvUnit2D/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/ConvUnit2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/ConvUnit2D/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_v2.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/calculate_wt_embedding_v2_kernel.cu
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/calculate_wt_embedding_v2_op.cpp
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/setup.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/wt_embedding_v2_ops.cpython-312-x86_64-linux-gnu.so
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Embedded/cuda_version_v2/include/utils.h
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v2/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v2/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_v3.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/calculate_wt_fc_v3_kernel.cu
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/calculate_wt_fc_v3_op.cpp
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/setup.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/wt_fc_v3_ops.cpython-312-x86_64-linux-gnu.so
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Linear_v3/cuda_version_v3/include/utils.h
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/MaxPool2D/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/MaxPool2D/cuda_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/MaxPool2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/MaxPool2D/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/MaxPool2D/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Padding/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Padding/original.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Padding/pytorch.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_v2.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_v3.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/pytorch_v2.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/calculate_wt_selfattention_v3_kernel.cu
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/calculate_wt_selfattention_v3_op.cpp
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/setup.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/wt_selfattention_v3_ops.cpython-312-x86_64-linux-gnu.so
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/SelfAttention/cuda_version_v3/include/utils.h
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/WtMaxunit2D/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/WtMaxunit2D/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/WtMaxunit2D/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/WtMaxunit2D/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_add_equal/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_add_equal/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_add_equal/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_add_equal/refactored_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_mul/__init__.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_mul/cuda_v1.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_mul/original_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_mul/pytorch_version.py
dl_backtrace/pytorch_backtrace/dlbacktrace/utils/cuda_utils/Wt_mul/refactored_version.py
docker/Dockerfile
docker/docker-compose.yml
docs/BUILD_DOCS.md
docs/CNAME
docs/index.md
docs/assets/css/custom.css
docs/assets/images/AryaXAI-light.jpg
docs/assets/images/AryaXAI.png
docs/assets/images/arya-fav.png
docs/assets/images/arya-logo-mini.png
docs/assets/images/dlb_logo.png
docs/assets/images/lexsi_logo.svg
docs/developer/architecture.md
docs/developer/contributing.md
docs/developer/cuda-development.md
docs/developer/execution-engine.md
docs/developer/testing.md
docs/examples/colab-notebooks.md
docs/examples/use-cases.md
docs/guide/best-practices.md
docs/guide/introduction.md
docs/guide/pytorch/auto-sampler.md
docs/guide/pytorch/dlbacktrace.md
docs/guide/pytorch/execution-engines.md
docs/guide/pytorch/moe-models.md
docs/guide/pytorch/operations.md
docs/guide/pytorch/overview.md
docs/guide/pytorch/pipeline.md
docs/guide/pytorch/temperature-scaling.md
docs/guide/pytorch/tracing.md
docs/guide/relevance/overview.md
docs/guide/relevance/parameters.md
docs/home/features.md
docs/home/installation.md
docs/home/quickstart.md
docs/home/whats-new.md
docs/javascripts/mathjax.js
docs/support/changelog.md
docs/support/faq.md
docs/support/known-issues.md
docs/support/license.md