Enable causal and move the persistent for backwards. (#704)

Summary:

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

Reviewed By: xuzhao9

Differential Revision: D88766359

Author: njriasan