Re-organize drifting LSSM demo
Add save/load functionality to VB inference