from colossalai.zero.shard_utils.base_shard_strategy import BaseShardStrategy
from colossalai.zero.shard_utils.tensor_shard_strategy import TensorShardStrategy

__all__ = ['BaseShardStrategy', 'TensorShardStrategy']