GitHub repository: trivikram-muralidharan/swa
Primary language: Python
Description: Stochastic Weight Averaging in PyTorch