RoMa: A lightweight library to deal with 3D rotations in PyTorch. - View it on GitHub