An updated version of DWE from 'Learning Wasserstein Embeddings' (https://arxiv.org/abs/1710.07457), now written in python3 with JAX & JAX-OTT.
This repository is part of the benchmarking effort for https://github.com/AnonWormhole/WormholeRebuttal, and can be installed by cloning the repository and using pip.