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/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/api/base_checkpointer.py
vescale/checkpoint/api/meta_type.py
vescale/checkpoint/api/vescale_checkpointer.py
vescale/checkpoint/planner/__init__.py
vescale/checkpoint/planner/common.py
vescale/checkpoint/planner/vescale/__init__.py
vescale/checkpoint/planner/vescale/vescale_planner.py
vescale/checkpoint/planner/vescale/vescale_planner_helpers.py
vescale/checkpoint/storage/filesystem.py
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/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/csrc/PLACEHOLDER
vescale/ddp/README.md
vescale/ddp/distributed_data_parallel.py
vescale/ddp/grad_buffer.py
vescale/debug/__init__.py
vescale/debug/debug_log.py
vescale/debug/pdb.py
vescale/devicemesh_api/README.md
vescale/devicemesh_api/__init__.py
vescale/devicemesh_api/api.py
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/dmp/README.md
vescale/dmp/__init__.py
vescale/dmp/dmp.py
vescale/dmp/policies/__init__.py
vescale/dmp/policies/megatron.py
vescale/dmp/policies/registry.py
vescale/dmp/policies/utils.py
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/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/initialize/__init__.py
vescale/initialize/deferred_init.py
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