LICENSE
README.md
pyproject.toml
spmd_types/__init__.py
spmd_types/_backward_hooks.py
spmd_types/_collectives.py
spmd_types/_dist.py
spmd_types/_dtensor.py
spmd_types/_dtensor_checker.py
spmd_types/_dtype_utils.py
spmd_types/_frame.py
spmd_types/_local.py
spmd_types/_mesh.py
spmd_types/_mesh_axis.py
spmd_types/_mesh_region.py
spmd_types/_raw_dist.py
spmd_types/_reinterpret_mesh.py
spmd_types/_scalar.py
spmd_types/_scalar_sentinel.py
spmd_types/_state.py
spmd_types/_test_utils.py
spmd_types/_testing.py
spmd_types/_traceback.py
spmd_types/_type_attr.py
spmd_types/_vmap.py
spmd_types/checker.py
spmd_types/runtime.py
spmd_types/types.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
spmd_types/_checker/__init__.py