#150 · Primary category: Deep Learning Frameworks

LibMTL

deep-learning mmoe mtl multi-domain-learning multi-objective-optimization multi-task-learning multiobjective-optimization multitask-learning ple python pytorch

A PyTorch Library for Multi-Task Learning

Project last updated:05/14/25

GitHub Stars

2.6K

Forks

232

Contributors

4

License

MIT

Why we included this project

Multi-task learning means training one model to predict several outputs from the same input, and LibMTL is a PyTorch library that takes care of the plumbing. Instead of writing a fresh training loop, you pick among the built-in published methods, which span both network designs and ways to combine losses or balance gradients, and get consistent preprocessing and evaluation on standard benchmarks so results compare fairly. Most methods are exposed as command-line flags, while the modular structure lets researchers plug in an experimental weighting scheme without rewriting the rest. The maintainers keep recent papers implemented, which makes the library a reasonable base whether you are building toward production or setting up the baselines for a research project.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category