LICENSE.apache-2.0
MANIFEST.in
README.md
requirements.txt
setup.cfg
setup.py
docs/_sources/index.rst.txt
docs/_sources/CONFIGURATION/activation_checkpointing.md.txt
docs/_sources/CONFIGURATION/kernel_fusion.md.txt
docs/_sources/CONFIGURATION/model_parallelism.md.txt
docs/_sources/TUTORIALS/activation_checkpointing.md.txt
docs/_sources/TUTORIALS/dataset.md.txt
docs/_sources/TUTORIALS/kernel_fusion.md.txt
docs/_sources/TUTORIALS/model_parallelism.md.txt
docs/_sources/TUTORIALS/tensor_model_data_parallelism.md.txt
docs/_sources/TUTORIALS/tensor_model_parallelism.md.txt
docs/_sources/TUTORIALS/tensor_model_zero_data_parallelism.md.txt
docs/_static/vendor/fontawesome/5.13.0/LICENSE.txt
docs/build/html/_sources/index.rst.txt
docs/build/html/_sources/CONFIGURATION/activation_checkpointing.md.txt
docs/build/html/_sources/CONFIGURATION/kernel_fusion.md.txt
docs/build/html/_sources/CONFIGURATION/model_parallelism.md.txt
docs/build/html/_sources/TUTORIALS/activation_checkpointing.md.txt
docs/build/html/_sources/TUTORIALS/dataset.md.txt
docs/build/html/_sources/TUTORIALS/kernel_fusion.md.txt
docs/build/html/_sources/TUTORIALS/model_parallelism.md.txt
docs/build/html/_sources/TUTORIALS/tensor_model_data_parallelism.md.txt
docs/build/html/_sources/TUTORIALS/tensor_model_parallelism.md.txt
docs/build/html/_sources/TUTORIALS/tensor_model_zero_data_parallelism.md.txt
docs/build/html/_static/vendor/fontawesome/5.13.0/LICENSE.txt
oslo/__init__.py
oslo/__version__.py
oslo/pytorch/__init__.py
oslo/pytorch/_C/__init__.py
oslo/pytorch/_C/csrc/CUDABinder.cpp
oslo/pytorch/_C/csrc/CompileCache.cpp
oslo/pytorch/_C/csrc/FusedLayerNorm.cu
oslo/pytorch/_C/csrc/FusedNoRepeatNGram.cu
oslo/pytorch/_C/csrc/__init__.py
oslo/pytorch/_C/csrc/includes/CompileCache.h
oslo/pytorch/_C/csrc/includes/compat.h
oslo/pytorch/_C/csrc/includes/type_shim.h
oslo/pytorch/activation_checkpointing/__init__.py
oslo/pytorch/activation_checkpointing/checkpoint_engine.py
oslo/pytorch/activation_checkpointing/utils/__init__.py
oslo/pytorch/activation_checkpointing/utils/checkpoint_function.py
oslo/pytorch/activation_checkpointing/utils/checkpoint_partitioner.py
oslo/pytorch/activation_checkpointing/utils/rng_state_tracker.py
oslo/pytorch/activation_checkpointing/utils/utils.py
oslo/pytorch/kernel_fusion/__init__.py
oslo/pytorch/kernel_fusion/kernel_fusion_engine.py
oslo/pytorch/kernel_fusion/cuda/__init__.py
oslo/pytorch/kernel_fusion/cuda/engine.py
oslo/pytorch/kernel_fusion/cuda/fused_ngram_blocking.py
oslo/pytorch/kernel_fusion/cuda/fused_normalization.py
oslo/pytorch/kernel_fusion/jit_partial/__init__.py
oslo/pytorch/kernel_fusion/jit_partial/engine.py
oslo/pytorch/kernel_fusion/jit_partial/fused_gelu.py
oslo/pytorch/kernel_fusion/mem_efficient/__init__.py
oslo/pytorch/kernel_fusion/mem_efficient/aot_autograd.py
oslo/pytorch/kernel_fusion/mem_efficient/compilers.py
oslo/pytorch/kernel_fusion/mem_efficient/decompositions.py
oslo/pytorch/kernel_fusion/mem_efficient/engine.py
oslo/pytorch/kernel_fusion/mem_efficient/model_output.py
oslo/pytorch/kernel_fusion/mem_efficient/partitioners.py
oslo/pytorch/kernel_fusion/mem_efficient/python_key.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/__init__.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/_stateless.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/aot_autograd.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/decompositions.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/partitioners.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/python_key.py
oslo/pytorch/kernel_fusion/mem_efficient/compat/shape_prop.py
oslo/pytorch/kernel_fusion/utils/__init__.py
oslo/pytorch/kernel_fusion/utils/torch_version.py
oslo/pytorch/model_parallelism/__init__.py
oslo/pytorch/model_parallelism/tensor_parallel_enigne.py
oslo/pytorch/model_parallelism/network/__init__.py
oslo/pytorch/model_parallelism/network/mpu.py
oslo/pytorch/model_parallelism/utils/__init__.py
oslo/pytorch/model_parallelism/utils/distributed.py
oslo/pytorch/model_parallelism/utils/extensions.py
oslo/pytorch/model_parallelism/utils/mappings.py
oslo/pytorch/utils/__init__.py
oslo/pytorch/utils/extenstions.py
oslo_core.egg-info/PKG-INFO
oslo_core.egg-info/SOURCES.txt
oslo_core.egg-info/dependency_links.txt
oslo_core.egg-info/not-zip-safe
oslo_core.egg-info/requires.txt
oslo_core.egg-info/top_level.txt