Implementation of Danijar's latest iteration for his Dreamer line of work - View it on GitHub