torch
numpy
arm-pytorch-utilities
pytorch-kinematics

[test]
pytest
