#150 · Primary category: Deep Learning Frameworks
LibMTL
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
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.