PyTorch models
=========================

PyTorch-based CATE estimators

.. toctree:: 
    :glob:
    :maxdepth: 2

    T-Learners <generated/catenets.models.torch.tlearner.rst>
    S-Learners <generated/catenets.models.torch.slearner.rst>
    Pseudo-Outcome Nets <generated/catenets.models.torch.pseudo_outocome_nets.rst>
    Representation Nets <generated/catenets.models.torch.representation_nets.rst>
    S-Nets <generated/catenets.models.torch.snet.rst>
