Thanks for sharing! Replica Exchange HMC is highly underappreciated and it is lovely to see a nice product build on top of it.
Not sure how much you can share, but I am curious about the where the sampling actually happen (ie how the RE-HMC is implemented). Since both PyMC and Stan model are accepted, I am assuming the implementation is written in something easy to compiled to run on multiple device?