I went through this in a video using the paper's official code - and it worked fairly well!

Definitely a great step forward in terms of reasoning tasks - even if it is an expensive step.