They are using smooth time integration, where I am using the discrete scheme described in the arxiv paper. The switch isn't that difficult though, since it just involves modifying a single line of the shader.