Contents Menu Expand Light mode Dark mode Auto light/dark mode
TorchJD
Light Logo Dark Logo

Getting Started

  • Installation
  • Examples
    • Basic Usage
    • Instance-Wise Risk Minimization (IWRM)
    • Multi-Task Learning (MTL)
    • Recurrent Neural Network (RNN)
    • PyTorch Lightning Integration
    • Automatic Mixed Precision (AMP)

API Reference

  • backward
  • mtl_backward
  • Aggregation
    • Aggregator (abstract)
    • UPGrad
    • Aligned-MTL
    • CAGrad
    • ConFIG
    • Constant
    • DualProj
    • GradDrop
    • IMTL-G
    • Krum
    • Mean
    • MGDA
    • Nash-MTL
    • PCGrad
    • Random
    • Sum
    • Trimmed Mean
Back to top
Copyright © Valerian Rey, Pierre Quinton
Made with Sphinx and @pradyunsg's Furo