cdopt.manifold_torch contains predefined manifold classes that choose PyTorch as their backbones.
cdopt.manifold_torch