LICENSE
NOTICE
README.md
pyproject.toml
setup.py
vescale/__init__.py
vescale.egg-info/PKG-INFO
vescale.egg-info/SOURCES.txt
vescale.egg-info/dependency_links.txt
vescale.egg-info/top_level.txt
vescale/__pycache__/__init__.cpython-39.pyc
vescale/checkpoint/README.md
vescale/checkpoint/__init__.py
vescale/checkpoint/load_state_dict.py
vescale/checkpoint/save_state_dict.py
vescale/checkpoint/version.py
vescale/checkpoint/__pycache__/__init__.cpython-39.pyc
vescale/checkpoint/__pycache__/load_state_dict.cpython-39.pyc
vescale/checkpoint/__pycache__/save_state_dict.cpython-39.pyc
vescale/checkpoint/api/base_checkpointer.py
vescale/checkpoint/api/meta_type.py
vescale/checkpoint/api/vescale_checkpointer.py
vescale/checkpoint/api/__pycache__/base_checkpointer.cpython-39.pyc
vescale/checkpoint/api/__pycache__/meta_type.cpython-39.pyc
vescale/checkpoint/api/__pycache__/vescale_checkpointer.cpython-39.pyc
vescale/checkpoint/planner/__init__.py
vescale/checkpoint/planner/common.py
vescale/checkpoint/planner/__pycache__/__init__.cpython-39.pyc
vescale/checkpoint/planner/__pycache__/common.cpython-39.pyc
vescale/checkpoint/planner/vescale/__init__.py
vescale/checkpoint/planner/vescale/vescale_planner.py
vescale/checkpoint/planner/vescale/vescale_planner_helpers.py
vescale/checkpoint/planner/vescale/__pycache__/__init__.cpython-39.pyc
vescale/checkpoint/planner/vescale/__pycache__/vescale_planner.cpython-39.pyc
vescale/checkpoint/planner/vescale/__pycache__/vescale_planner_helpers.cpython-39.pyc
vescale/checkpoint/storage/filesystem.py
vescale/checkpoint/storage/__pycache__/filesystem.cpython-39.pyc
vescale/checkpoint/utilities/bfile.py
vescale/checkpoint/utilities/logger.py
vescale/checkpoint/utilities/mem_checkpoint.py
vescale/checkpoint/utilities/sync_queue.py
vescale/checkpoint/utilities/__pycache__/bfile.cpython-39.pyc
vescale/checkpoint/utilities/__pycache__/logger.cpython-39.pyc
vescale/checkpoint/utilities/__pycache__/mem_checkpoint.cpython-39.pyc
vescale/checkpoint/utilities/server/__init__.py
vescale/checkpoint/utilities/server/detached_mem_server.py
vescale/checkpoint/utilities/server/mem_file_service.proto
vescale/checkpoint/utilities/server/mem_file_service_pb2.py
vescale/checkpoint/utilities/server/mem_file_service_pb2.pyi
vescale/checkpoint/utilities/server/mem_file_service_pb2_grpc.py
vescale/checkpoint/utilities/server/mem_server_lib.py
vescale/checkpoint/utilities/server/report_service.proto
vescale/checkpoint/utilities/server/report_service_pb2.py
vescale/checkpoint/utilities/server/report_service_pb2.pyi
vescale/checkpoint/utilities/server/report_service_pb2_grpc.py
vescale/checkpoint/utilities/server/server_lib.py
vescale/checkpoint/utilities/server/server_status_client.py
vescale/checkpoint/utilities/server/__pycache__/__init__.cpython-39.pyc
vescale/checkpoint/utilities/server/__pycache__/mem_file_service_pb2.cpython-39.pyc
vescale/checkpoint/utilities/server/__pycache__/mem_file_service_pb2_grpc.cpython-39.pyc
vescale/checkpoint/utilities/server/__pycache__/mem_server_lib.cpython-39.pyc
vescale/csrc/PLACEHOLDER
vescale/ddp/README.md
vescale/ddp/distributed_data_parallel.py
vescale/ddp/grad_buffer.py
vescale/ddp/__pycache__/distributed_data_parallel.cpython-39.pyc
vescale/ddp/__pycache__/grad_buffer.cpython-39.pyc
vescale/debug/__init__.py
vescale/debug/debug_log.py
vescale/debug/pdb.py
vescale/debug/__pycache__/__init__.cpython-39.pyc
vescale/debug/__pycache__/debug_log.cpython-39.pyc
vescale/devicemesh_api/README.md
vescale/devicemesh_api/__init__.py
vescale/devicemesh_api/api.py
vescale/devicemesh_api/__pycache__/__init__.cpython-39.pyc
vescale/devicemesh_api/__pycache__/api.cpython-39.pyc
vescale/dmodule/README.md
vescale/dmodule/__init__.py
vescale/dmodule/_dmodule.py
vescale/dmodule/_factory.py
vescale/dmodule/_grad_sync.py
vescale/dmodule/_hook.py
vescale/dmodule/api.py
vescale/dmodule/placements_interface.py
vescale/dmodule/__pycache__/__init__.cpython-39.pyc
vescale/dmodule/__pycache__/_dmodule.cpython-39.pyc
vescale/dmodule/__pycache__/_grad_sync.cpython-39.pyc
vescale/dmodule/__pycache__/_hook.cpython-39.pyc
vescale/dmodule/__pycache__/api.cpython-39.pyc
vescale/dmodule/__pycache__/placements_interface.cpython-39.pyc
vescale/dmp/README.md
vescale/dmp/__init__.py
vescale/dmp/dmp.py
vescale/dmp/__pycache__/__init__.cpython-39.pyc
vescale/dmp/__pycache__/dmp.cpython-39.pyc
vescale/dmp/policies/__init__.py
vescale/dmp/policies/megatron.py
vescale/dmp/policies/registry.py
vescale/dmp/policies/utils.py
vescale/dmp/policies/__pycache__/__init__.cpython-39.pyc
vescale/dmp/policies/__pycache__/megatron.cpython-39.pyc
vescale/dmp/policies/__pycache__/registry.cpython-39.pyc
vescale/dmp/policies/__pycache__/utils.cpython-39.pyc
vescale/dtensor/README.md
vescale/dtensor/__init__.py
vescale/dtensor/_collective_utils.py
vescale/dtensor/_diff.py
vescale/dtensor/_dispatch_bypass.py
vescale/dtensor/_dispatch_patch.py
vescale/dtensor/_dynamo_utils.py
vescale/dtensor/_utils.py
vescale/dtensor/api.py
vescale/dtensor/device_mesh.py
vescale/dtensor/dispatch.py
vescale/dtensor/dtensor.py
vescale/dtensor/loss.py
vescale/dtensor/op_schema.py
vescale/dtensor/placement_types.py
vescale/dtensor/random.py
vescale/dtensor/redistribute.py
vescale/dtensor/sharding_prop.py
vescale/dtensor/__pycache__/__init__.cpython-39.pyc
vescale/dtensor/__pycache__/_collective_utils.cpython-39.pyc
vescale/dtensor/__pycache__/_diff.cpython-39.pyc
vescale/dtensor/__pycache__/_dispatch_bypass.cpython-39.pyc
vescale/dtensor/__pycache__/_dispatch_patch.cpython-39.pyc
vescale/dtensor/__pycache__/_dynamo_utils.cpython-39.pyc
vescale/dtensor/__pycache__/_utils.cpython-39.pyc
vescale/dtensor/__pycache__/api.cpython-39.pyc
vescale/dtensor/__pycache__/device_mesh.cpython-39.pyc
vescale/dtensor/__pycache__/dispatch.cpython-39.pyc
vescale/dtensor/__pycache__/dtensor.cpython-39.pyc
vescale/dtensor/__pycache__/op_schema.cpython-39.pyc
vescale/dtensor/__pycache__/placement_types.cpython-39.pyc
vescale/dtensor/__pycache__/random.cpython-39.pyc
vescale/dtensor/__pycache__/redistribute.cpython-39.pyc
vescale/dtensor/__pycache__/sharding_prop.cpython-39.pyc
vescale/dtensor/ops/__init__.py
vescale/dtensor/ops/basic_strategy.py
vescale/dtensor/ops/common_rules.py
vescale/dtensor/ops/conv_ops.py
vescale/dtensor/ops/embedding_ops.py
vescale/dtensor/ops/experimental_ops.py
vescale/dtensor/ops/math_ops.py
vescale/dtensor/ops/matrix_ops.py
vescale/dtensor/ops/pointwise_ops.py
vescale/dtensor/ops/random_ops.py
vescale/dtensor/ops/tensor_ops.py
vescale/dtensor/ops/utils.py
vescale/dtensor/ops/vescale_view_ops.py
vescale/dtensor/ops/view_ops.py
vescale/dtensor/ops/__pycache__/__init__.cpython-39.pyc
vescale/dtensor/ops/__pycache__/basic_strategy.cpython-39.pyc
vescale/dtensor/ops/__pycache__/common_rules.cpython-39.pyc
vescale/dtensor/ops/__pycache__/conv_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/embedding_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/experimental_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/math_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/matrix_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/pointwise_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/random_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/tensor_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/utils.cpython-39.pyc
vescale/dtensor/ops/__pycache__/vescale_view_ops.cpython-39.pyc
vescale/dtensor/ops/__pycache__/view_ops.cpython-39.pyc
vescale/initialize/__init__.py
vescale/initialize/deferred_init.py
vescale/initialize/__pycache__/__init__.cpython-39.pyc
vescale/initialize/__pycache__/deferred_init.cpython-39.pyc
vescale/model/__init__.py
vescale/model/patch/__init__.py
vescale/model/patch/linear.py
vescale/model/patch/utils.py
vescale/model/patch/vp_cross_entropy.py
vescale/model/patch/vp_embedding.py
vescale/optim/README.md
vescale/optim/base_optimizer.py
vescale/optim/clip_grads.py
vescale/optim/distributed_optimizer.py
vescale/optim/utils.py
vescale/optim/__pycache__/base_optimizer.cpython-39.pyc
vescale/optim/__pycache__/clip_grads.cpython-39.pyc
vescale/optim/__pycache__/distributed_optimizer.cpython-39.pyc
vescale/optim/__pycache__/utils.cpython-39.pyc