PyTorch re-implementation of [Structured Inference Networks for Nonlinear State Space Models, AAAI 17]