LICENSE
README.md
setup.py
megatron/core/__init__.py
megatron/core/enums.py
megatron/core/package_info.py
megatron/core/parallel_state.py
megatron/core/utils.py
megatron/core/pipeline_parallel/__init__.py
megatron/core/pipeline_parallel/p2p_communication.py
megatron/core/pipeline_parallel/schedules.py
megatron/core/tensor_parallel/__init__.py
megatron/core/tensor_parallel/cross_entropy.py
megatron/core/tensor_parallel/data.py
megatron/core/tensor_parallel/layers.py
megatron/core/tensor_parallel/mappings.py
megatron/core/tensor_parallel/random.py
megatron/core/tensor_parallel/utils.py
megatron_core.egg-info/PKG-INFO
megatron_core.egg-info/SOURCES.txt
megatron_core.egg-info/dependency_links.txt
megatron_core.egg-info/requires.txt
megatron_core.egg-info/top_level.txt