Use state of the art density estimation for an RL dynamics model.
action
state
next state
Density estimate