https://medium.com/@monocosmo77/working-with-bures-wasserstein-metric-part10-machine-learning-2024-3f36b6e9786c