torch
se3-transformer