* Use pyro.checkpoint