This is my master's research project, supervised by Hippolyt Ritter and David Barber at the UCL Centre for AI. If interested I recommend reading the introductory chapter of my report (report_localglobal.pdf) to get a high-level idea.

Codebase references:

https://github.com/wjmaddox/swa_gaussian

https://github.com/timgaripov/swa/

https://github.com/izmailovpavel/understandingbdl/