https://hunterheidenreich.com/posts/kabsch-algorithm/
Kabsch Algorithm: NumPy, PyTorch, TensorFlow, and JAX