BatchCentering and BatchLipNorm classes compatible with multi-GPU - #55

Merged

franckma31 merged 19 commits into

Mar 23, 2026

Commits on Jan 22, 2026

modify BatchCentering to support multi-GPU training, based on a running_mean accumulator and a running_num_batches
franckma31
committed
e127509

Commits on Jan 24, 2026

refactoring to prepare BatchLipNorm merging
franckma31
committed
12908e3

Commits on Jan 25, 2026

add Vanilla export for BatchCentering with the layer ScaleBiasLayer
franckma31
committed
663a92d

Commits on Jan 26, 2026

Commits on Jan 27, 2026

Add support for BatchLipNorm computing the variance on batch and full set
franckma31
committed
44fe2d9

Commits on Jan 28, 2026

Commits on Jan 30, 2026

Commits on Feb 2, 2026

modify running variable update to avoid out of graph modifications
franckma31
committed
abd3bd8

Commits on Mar 16, 2026

Merge pull request #54 from deel-ai/feature/batchlipnorm
franckma31
authored
bfd01fd