pytorch_xd

PyTorch_xd是一个高等级的PyTorch训练辅助库。