PyTorch4D

A set of Pytorch compatible modules for use with 4D tensors.