robotics python3 inverse-kinematics manifold-optimization pytorch-implementation transformer-encoder pydantic position-embedding pytorch-lightning keypointnet multihead-self-attention visual-transformer transfomers dense-object-nets furthest-point-sampling mvit