.codespell-whitelist.txt
.gitignore
.pre-commit-config.yaml
LICENSE
LICENSE.header
README.md
anaconda-project.yml
licensing.sh
pyproject.toml
requirements.txt
setup.py
tox.ini
.github/merge_rules.yaml
.github/workflows/push.yaml
LICENSES/LICENSE.AGPL
LICENSES/LICENSE.Apache
LICENSES/LICENSE.BSD
LICENSES/LICENSE.GPLv2
LICENSES/LICENSE.GPLv3
LICENSES/LICENSE.MIT
LICENSES/LICENSE.Unlicense
danling/__init__.py
danling/_version.py
danling/defaults.py
danling/typing.py
danling.egg-info/PKG-INFO
danling.egg-info/SOURCES.txt
danling.egg-info/dependency_links.txt
danling.egg-info/requires.txt
danling.egg-info/top_level.txt
danling/data/__init__.py
danling/data/dataloader.py
danling/data/loader_wrappers.py
danling/data/utils.py
danling/metrics/README.md
danling/metrics/__init__.py
danling/metrics/average_meter.py
danling/metrics/factory.py
danling/metrics/global_metrics.py
danling/metrics/multitask.py
danling/metrics/preprocess.py
danling/metrics/registry.py
danling/metrics/state.py
danling/metrics/stream_metrics.py
danling/metrics/utils.py
danling/metrics/functional/__init__.py
danling/metrics/functional/binary.py
danling/metrics/functional/classification.py
danling/metrics/functional/multiclass.py
danling/metrics/functional/multilabel.py
danling/metrics/functional/regression.py
danling/metrics/functional/utils.py
danling/modules/__init__.py
danling/modules/mlp/__init__.py
danling/modules/mlp/dense.py
danling/modules/mlp/mlp.py
danling/modules/transformer/__init__.py
danling/modules/transformer/decoder.py
danling/modules/transformer/encoder.py
danling/modules/transformer/attention/__init__.py
danling/modules/transformer/attention/multihead_attention.py
danling/modules/transformer/attention/simple_attention.py
danling/modules/transformer/ffn/__init__.py
danling/modules/transformer/ffn/fcn.py
danling/modules/transformer/pos_embed/__init__.py
danling/modules/transformer/pos_embed/pos_embed.py
danling/optim/__init__.py
danling/optim/optimizer.py
danling/optim/registry.py
danling/optim/lr_scheduler/__init__.py
danling/optim/lr_scheduler/lr_scheduler.py
danling/optim/lr_scheduler/registry.py
danling/runners/README.md
danling/runners/__init__.py
danling/runners/base_runner.py
danling/runners/compile.py
danling/runners/config.py
danling/runners/deepspeed_runner.py
danling/runners/fault_tolerance.py
danling/runners/fsdp.py
danling/runners/graph_runner.py
danling/runners/parallel_runner.py
danling/runners/runner.py
danling/runners/state.py
danling/runners/supervisor.py
danling/runners/telemetry.py
danling/runners/topology.py
danling/runners/torch_runner.py
danling/runners/utils.py
danling/runners/workspace.py
danling/runners/checkpoints/__init__.py
danling/runners/checkpoints/base.py
danling/runners/checkpoints/file.py
danling/runners/checkpoints/torch_distributed.py
danling/runners/checkpoints/torch_ft.py
danling/runners/mixins/__init__.py
danling/runners/mixins/fp8.py
danling/tensors/README.md
danling/tensors/__init__.py
danling/tensors/aten_functions.py
danling/tensors/nested_tensor.py
danling/tensors/nn_functions.py
danling/tensors/ops.py
danling/tensors/pn_tensor.py
danling/tensors/torch_functions.py
danling/tensors/functions/__init__.py
danling/tensors/functions/_convolution.py
danling/tensors/functions/_pooling.py
danling/tensors/functions/channel.py
danling/tensors/functions/conv1d.py
danling/tensors/functions/conv2d.py
danling/tensors/functions/conv_transpose1d.py
danling/tensors/functions/conv_transpose2d.py
danling/tensors/functions/pool1d.py
danling/tensors/functions/pool2d.py
danling/tensors/functions/pool3d.py
danling/utils/__init__.py
danling/utils/basex.py
danling/utils/context_managers.py
danling/utils/decorators.py
danling/utils/descriptors.py
danling/utils/distributed.py
danling/utils/io.py
danling/utils/lists.py
danling/utils/round_dict.py
docs/mkdocs.yml
docs/requirements.txt
docs/docs/CNAME
docs/docs/index.md
docs/docs/manifest.webmanifest
docs/docs/package.md
docs/docs/about/index.md
docs/docs/about/index.zh.md
docs/docs/about/license.md
docs/docs/about/license.zh.md
docs/docs/about/privacy.md
docs/docs/about/privacy.zh.md
docs/docs/blog/index.md
docs/docs/metrics/average_meter.md
docs/docs/metrics/factory.md
docs/docs/metrics/functional.md
docs/docs/metrics/index.md
docs/docs/metrics/metric_meter.md
docs/docs/metrics/metrics.md
docs/docs/metrics/utils.md
docs/docs/optim/lr_scheduler.md
docs/docs/runners/base_runner.md
docs/docs/runners/config.md
docs/docs/runners/index.md
docs/docs/runners/runner.md
docs/docs/runners/torch_runner.md
docs/docs/runners/utils.md
docs/docs/tensors/functions.md
docs/docs/tensors/index.md
docs/docs/tensors/nested_tensor.md
docs/docs/tensors/pn_tensor.md
docs/docs/tensors/utils.md
docs/docs/utils/basex.md
docs/docs/utils/context_managers.md
docs/docs/utils/decorators.md
docs/docs/utils/io.md
docs/overrides/main.html
docs/overrides/assets/css/extra.css
docs/overrides/assets/css/fonts.css
docs/overrides/assets/fonts/CascadiaCodePL.woff2
docs/overrides/assets/fonts/HYQiHei.ttf
docs/overrides/assets/fonts/HelveticaNowDisplay.otf
docs/overrides/assets/fonts/HelveticaWorld.ttf
docs/overrides/assets/images/favicon.ico
docs/overrides/assets/images/logo.png
docs/overrides/javascripts/mathjax.js
docs/overrides/javascripts/shortcuts.js
examples/runners/imdb.py
examples/runners/mnist.py
examples/tensors/imdb.py
scripts/benchmark_nested_tensor.py
tests/__init__.py
tests/conftest.py
tests/examples/__init__.py
tests/examples/runners/__init__.py
tests/examples/runners/test_mnist.py
tests/metrics/__init__.py
tests/metrics/test_average_meter.py
tests/metrics/test_factory.py
tests/metrics/test_global_metrics.py
tests/metrics/test_multitask.py
tests/metrics/test_preprocess.py
tests/metrics/test_registry.py
tests/metrics/test_state.py
tests/metrics/test_stream_metrics.py
tests/metrics/test_utils.py
tests/metrics/utils.py
tests/metrics/functional/test_binary.py
tests/metrics/functional/test_classification.py
tests/metrics/functional/test_multiclass.py
tests/metrics/functional/test_multilabel.py
tests/metrics/functional/test_regression.py
tests/modules/mlp/test_mlp.py
tests/modules/transformer/test_decoder.py
tests/modules/transformer/test_encoder.py
tests/modules/transformer/attention/test_attention.py
tests/modules/transformer/ffn/test_fcn.py
tests/modules/transformer/pos_embed/test_pos_embed.py
tests/optim/test_lr_scheduler.py
tests/optim/test_optimizers.py
tests/optim/test_schedulers.py
tests/runners/__init__.py
tests/runners/distributed.py
tests/runners/test_base_runner.py
tests/runners/test_compile.py
tests/runners/test_config.py
tests/runners/test_deepspeed_runner.py
tests/runners/test_fault_tolerance.py
tests/runners/test_fsdp.py
tests/runners/test_graph_runner.py
tests/runners/test_parallel_runner.py
tests/runners/test_runner.py
tests/runners/test_state.py
tests/runners/test_supervisor.py
tests/runners/test_telemetry.py
tests/runners/test_topology.py
tests/runners/test_torch_runner.py
tests/runners/test_utils.py
tests/runners/test_workspace.py
tests/runners/checkpoints/test_file.py
tests/runners/checkpoints/test_torch_distributed.py
tests/runners/integration/__init__.py
tests/runners/integration/test_fault_tolerance.py
tests/runners/integration/test_supervisor.py
tests/runners/integration/workers/fault_tolerance_torch_worker.py
tests/runners/integration/workers/fault_tolerance_worker.py
tests/runners/integration/workers/sigterm_torch_worker.py
tests/tensors/__init__.py
tests/tensors/test_aten_functions.py
tests/tensors/test_nested_tensor.py
tests/tensors/test_nn_functions.py
tests/tensors/test_ops.py
tests/tensors/test_pn_tensor.py
tests/tensors/test_torch_functions.py
tests/tensors/utils.py
tests/tensors/functions/test_channel.py
tests/tensors/functions/test_conv1d.py
tests/tensors/functions/test_conv2d.py
tests/tensors/functions/test_conv_transpose1d.py
tests/tensors/functions/test_conv_transpose2d.py
tests/tensors/functions/test_pool1d.py
tests/tensors/functions/test_pool2d.py
tests/tensors/functions/test_pool3d.py
tests/utils/test_decorators.py
tests/utils/test_descriptors.py
tests/utils/test_round_dict.py