Allow batch computations for jacobian with pytorch

[evelyd]

Made the necessary changes to allow batch computations with pytorch (did not test other modalities) for the jacobian, forward kinematics, and all the functions used by those.