NVIDIA's Deep Imagination Team's PyTorch Library - View it on GitHub