LICENSE
MANIFEST.in
README.md
setup.py
lightning_thunder.egg-info/PKG-INFO
lightning_thunder.egg-info/SOURCES.txt
lightning_thunder.egg-info/dependency_links.txt
lightning_thunder.egg-info/not-zip-safe
lightning_thunder.egg-info/requires.txt
lightning_thunder.egg-info/top_level.txt
requirements/base.txt
requirements/devel.txt
requirements/docs.txt
requirements/notebooks.txt
requirements/test.txt
thunder/__about__.py
thunder/__init__.py
thunder/common.py
thunder/py.typed
thunder/benchmarks/__init__.py
thunder/benchmarks/benchmark_hf.py
thunder/benchmarks/benchmark_litgpt.py
thunder/benchmarks/benchmark_peft.py
thunder/benchmarks/conftest.py
thunder/benchmarks/distributed.py
thunder/benchmarks/einsum.py
thunder/benchmarks/targets.py
thunder/benchmarks/test_benchmark_litgpt.py
thunder/benchmarks/utils.py
thunder/clang/__init__.py
thunder/clang/langctx.py
thunder/core/__init__.py
thunder/core/baseutils.py
thunder/core/codeutils.py
thunder/core/compile_data.py
thunder/core/devices.py
thunder/core/dtypes.py
thunder/core/interpreter.py
thunder/core/jit_ext.py
thunder/core/langctxs.py
thunder/core/module.py
thunder/core/options.py
thunder/core/patterns.py
thunder/core/prims.py
thunder/core/profile.py
thunder/core/proxies.py
thunder/core/pytree.py
thunder/core/recipe.py
thunder/core/rematerialization.py
thunder/core/symbol.py
thunder/core/trace.py
thunder/core/trace_interpreter.py
thunder/core/transform_common.py
thunder/core/transforms.py
thunder/core/update_aliases.py
thunder/core/utils.py
thunder/core/vjp_utils.py
thunder/dev_utils/__init__.py
thunder/dev_utils/benchmark.py
thunder/dev_utils/check_trace.py
thunder/dev_utils/debug_transform.py
thunder/dev_utils/nvtx_profile_transform.py
thunder/dev_utils/utils.py
thunder/distributed/__init__.py
thunder/distributed/bucketing.py
thunder/distributed/checkpoint.py
thunder/distributed/prims.py
thunder/distributed/utils.py
thunder/distributed/tensor_parallel/__init__.py
thunder/distributed/tensor_parallel/column_wise.py
thunder/distributed/tensor_parallel/common.py
thunder/distributed/tensor_parallel/optimize_comm.py
thunder/distributed/tensor_parallel/row_wise.py
thunder/distributed/transforms/__init__.py
thunder/distributed/transforms/ddp.py
thunder/distributed/transforms/ddp_v2.py
thunder/distributed/transforms/fsdp.py
thunder/distributed/transforms/fsdp_v2.py
thunder/dynamo/__init__.py
thunder/dynamo/benchmark_utils.py
thunder/dynamo/compiler.py
thunder/dynamo/compiler_graph_benchmark.py
thunder/dynamo/report.py
thunder/dynamo/repro_script_template.py
thunder/dynamo/splitter.py
thunder/dynamo/utils.py
thunder/examine/__init__.py
thunder/examine/memory_calculation.py
thunder/executors/__init__.py
thunder/executors/apex_entropyex_impl.py
thunder/executors/apex_fused_rms_norm_impl.py
thunder/executors/apexex.py
thunder/executors/cudnn_layernormex.py
thunder/executors/cudnn_sdpa.py
thunder/executors/cudnnex.py
thunder/executors/data_dependent_partition.py
thunder/executors/fa3ex.py
thunder/executors/nvfuserex.py
thunder/executors/nvfuserex_impl.py
thunder/executors/passes.py
thunder/executors/pythonex.py
thunder/executors/sdpaex.py
thunder/executors/torch_autograd.py
thunder/executors/torch_compile.py
thunder/executors/torchex.py
thunder/executors/transformer_engine_v2ex.py
thunder/executors/transformer_engine_v2ex_impl.py
thunder/executors/transformer_engineex.py
thunder/executors/triton_crossentropy.py
thunder/executors/triton_crossentropy_impl.py
thunder/executors/triton_utils.py
thunder/executors/utils.py
thunder/extend/__init__.py
thunder/numpy/__init__.py
thunder/numpy/langctx.py
thunder/plugins/__init__.py
thunder/plugins/distributed.py
thunder/plugins/fp8.py
thunder/plugins/quantization.py
thunder/plugins/reduce_overhead.py
thunder/recipes/__init__.py
thunder/recipes/base.py
thunder/recipes/hf_transformers.py
thunder/torch/__init__.py
thunder/torch/default_torch_ops.py
thunder/torch/langctx.py
thunder/torch/experimental/__init__.py
thunder/torch/experimental/dtensor_codeutils.py
thunder/torch/experimental/dtensor_proxy.py
thunder/torch/experimental/dtensor_torch_and_prims.py
thunder/torch/experimental/dtensor_utils.py
thunder/transforms/__init__.py
thunder/transforms/autocast.py
thunder/transforms/autodiff.py
thunder/transforms/constant_folding.py
thunder/transforms/cudagraph.py
thunder/transforms/extraction_only_prologue_transform.py
thunder/transforms/materialization.py
thunder/transforms/prune_prologue_checks.py
thunder/transforms/qlora.py
thunder/transforms/quantization.py
thunder/transforms/utils.py