https://medium.com/@monocosmo77/working-with-bures-wasserstein-metric-part8-machine-learning-2024-8f78e8f0e9f6