kolmogorov_smirnov_torch A simple, vectorized implementation of the two-sample Kolmogorov-Smirnov test in PyTorch. Runs in $O(n_1+n_2)$ time, where $n_1$ and $n_2$ are the sizes of the two samples.