We chose DreamerV2 [1]}, a state-of-the-art MBRL approach for discrete environments, as a backbone for our algorithm. Following Dreamer notation, the model consists of six components:

RSSM={ll
Recurrent model:  hit = f(hit-1, zt-1, at-1)
