Tensor Shard Node Handlers (node_handler/__init__.py)

Handles tensor sharding logic for various neural network operations during distributed model parallelization. Use this when implementing custom parallelization strategies for deep

Repository

ColossalAI

Language

python

SPDX licence

Apache-2.0

Quality score (dated snapshot)

10/10

Category

ai

Review source and licence terms before reuse. View on the service