GitHub - khanrc/swad: Official Implementation of SWAD (NeurIPS 2021)