README.md
README.rst
pyproject.toml
setup.py
detrain/__init__.py
detrain.egg-info/PKG-INFO
detrain.egg-info/SOURCES.txt
detrain.egg-info/dependency_links.txt
detrain.egg-info/requires.txt
detrain.egg-info/top_level.txt
detrain/fsdp_tp/__init__.py
detrain/fsdp_tp/evaluation.py
detrain/fsdp_tp/log_utils.py
detrain/fsdp_tp/mesh_utils.py
detrain/fsdp_tp/model_2d.py
detrain/fsdp_tp/train.py
detrain/fsdp_tp/train_eval.py
detrain/ppl/__init__.py
detrain/ppl/args_util.py
detrain/ppl/dataset_util.py
detrain/ppl/dis_model.py
detrain/ppl/evaluation.py
detrain/ppl/master_node.py
detrain/ppl/train.py
detrain/ppl/worker.py
detrain/tp/__init__.py
detrain/tp/evaluation.py
detrain/tp/model_utils.py
detrain/tp/sequence_evaluation.py
detrain/tp/sequence_train.py
detrain/tp/sequence_train_eval.py
detrain/tp/train.py
detrain/tp/train_eval.py
examples/fsdp_tp/base_model.py
examples/fsdp_tp/main.py
examples/ppl/nn/base_model.py
examples/ppl/nn/main.py
examples/ppl/nn/shards_model.py
examples/ppl/restnet/base_model.py
examples/ppl/restnet/main.py
examples/ppl/restnet/shards_model.py
examples/tp/nn/base_model.py
examples/tp/nn/main.py
examples/tp/nn/rdz.py
examples/tp/nn/sequence_main.py
tests/test_dataset.py