[ANN] SlicedWasserstein.jl — Sliced Wasserstein distances and barycenters in Julia

Community
,
1

Hi everyone,

I’m happy to announce SlicedWasserstein.jl, a Julia package implementing:

• Sliced Wasserstein distance
• Sliced OT (SOT)
• Barycenters
• Sliced-Wasserstein K-means

The package is designed for discrete measures and uses Monte Carlo projections.

Example:

using SlicedWasserstein

μ = DiscreteMeasure(randn(2, 50))
ν = DiscreteMeasure(randn(2, 50))

sot = SOT(μ, ν)
println(sot)

Repository:

Feedback is very welcome!