Companion code to the paper "Variational Inference for Continuous-Time Switching Dynamical Systems" published in NeurIPS 2021.