Enable causal and move the persistent for backwards.

Summary: Applies the changes from the TLX tutorial to move backwards to use a persistent implementation and add initial causal support.

Differential Revision: D88766359

Author: njriasan