My implementation: https://github.com/keyehzy/thinking_with_looped_flows