Yes. I omitted time-indices, and moved the momentum-update up to before the parameter-update. In their pseudo-code θₜ updated with mₜ₋₁ whereas above, it would be with mₜ.
I suspect it is equivalent up to slightly different α and β?
I suspect it is equivalent up to slightly different α and β?