Had to refactor a bit to get a sensible system for loading weights asynchronously. I'm still not sure it is really sensible, but it works.