Module distributed

Module distributed 

Source
Expand description

Distributed Training

Support for multi-GPU training with data and model parallelism.

Structs§

DataParallel
Data parallel training
DistributedConfig
Distributed training configuration
DistributedDataParallel
Distributed Data Parallel (DDP)
GradientAccumulator
Gradient accumulation
ModelParallel
Model parallel training
PipelineParallel
Pipeline parallelism

Enums§

DistributedBackend
Distributed training backend