Scalable Approximation Algorithms for $p$-Wasserstein Distance and Its Variants