pytorch3d.implicitron.models.implicit_function