use scipy.spatial.transform.Rotation in rotations
The scipy.transform.spatial.Rotation
functionality seems to be available from jax! So we should probably use it in the rotations instead of setting up our own rotation matrices by hand.
The scipy.transform.spatial.Rotation
functionality seems to be available from jax! So we should probably use it in the rotations instead of setting up our own rotation matrices by hand.