deepfold.distributed.model_parallel.initialize

deepfold.distributed.model_parallel.initialize(dap_size: int) None[source]

Initialize Dynamic Axial Parallelism (DAP).

Parameters:

dap_size – number of GPUs used in DAP group.