Annotated implementation of Zero-DP memory optimization for deep learning
nn.labml.ai