LICENCE
README.md
pyproject.toml
setup.cfg
src/tensor_parallel/__init__.py
src/tensor_parallel/communications.py
src/tensor_parallel/cross_device_ops.py
src/tensor_parallel/dispatch.py
src/tensor_parallel/factory.py
src/tensor_parallel/pretrained_model.py
src/tensor_parallel/sharding.py
src/tensor_parallel/slicer_wrapper.py
src/tensor_parallel/slicing_configs.py
src/tensor_parallel/tensor_parallel.py
src/tensor_parallel/utils.py
src/tensor_parallel.egg-info/PKG-INFO
src/tensor_parallel.egg-info/SOURCES.txt
src/tensor_parallel.egg-info/dependency_links.txt
src/tensor_parallel.egg-info/requires.txt
src/tensor_parallel.egg-info/top_level.txt
tests/test_basic.py
tests/test_factory.py
tests/test_integration.py
tests/test_saving.py
tests/test_transformers.py