https://medium.com/@monocosmo77/working-with-bures-wasserstein-metric-part9-machine-learning-2024-4dfcf7d01d4f