LICENSE
pyproject.toml
sixlib/__init__.py
sixlib/spmd_types/__init__.py
sixlib/spmd_types/_checker.py
sixlib/spmd_types/_collectives.py
sixlib/spmd_types/_dist.py
sixlib/spmd_types/_dtensor.py
sixlib/spmd_types/_frame.py
sixlib/spmd_types/_local.py
sixlib/spmd_types/_mesh.py
sixlib/spmd_types/_mesh_axis.py
sixlib/spmd_types/_raw_dist.py
sixlib/spmd_types/_scalar.py
sixlib/spmd_types/_state.py
sixlib/spmd_types/_test_utils.py
sixlib/spmd_types/_testing.py
sixlib/spmd_types/_traceback.py
sixlib/spmd_types/_type_attr.py
sixlib/spmd_types/_vmap.py
sixlib/spmd_types/api_test.py
sixlib/spmd_types/checker_test.py
sixlib/spmd_types/collectives_test.py
sixlib/spmd_types/dtensor_test.py
sixlib/spmd_types/global_spmd_test.py
sixlib/spmd_types/local_test.py
sixlib/spmd_types/mesh_axis_test.py
sixlib/spmd_types/py.typed
sixlib/spmd_types/traceback_test.py
sixlib/spmd_types/types.py
sixlib/spmd_types/types_test.py
sixlib/spmd_types/vmap_test.py
spmd_types.egg-info/PKG-INFO
spmd_types.egg-info/SOURCES.txt
spmd_types.egg-info/dependency_links.txt
spmd_types.egg-info/requires.txt
spmd_types.egg-info/top_level.txt