Does this work with any distributed PostgreSQL solutions like Yugabyte or Cockroach? Citus?
It works on top DocumentDB PostgreSQL extension: https://github.com/microsoft/documentdb I don't think it could be made compatible with Cockroach without a complete rewrite in Go. Not sure about Yugabyte, though.