@ekzhang1 Yep, I'm not sure which branch you're currently using, but I had to strip a few things to make it easier to upstream: Once the last two PRs land, I will re-add a few features to make it more robust and handle these cases.

@ekzhang1 Hi, @ekzhang1 glad you liked my project! If you want to check out some larger use cases, some colleagues of mine published and includes some more examples