Another cool thing, their paper mentions that, "The specific implementation is based on the python package torch-two-sample, modified to use numba for improved

Another cool thing, their paper mentions that, "The specific implementation is based on the python package torch-two-sample, modified to use numba for improved performance." ( https://www.cell.com/action/showPdf?pii=S0092-8674%2822%2900597-9 https://www.cell.com/action/showPdf?pii=S0092-8674%2822%2900... ) If you haven't checked out the numba package, definitely worth a look for custom numerical computing in Python! disclosure - have made a small contribution to the package.