Multi-Task Learning ML Frameworks

Frameworks, libraries, and optimization methods for training models on multiple related tasks simultaneously. Includes MTL architectures, loss weighting strategies, gradient aggregation, and Pareto optimization. Does NOT include single-task learning, weight space learning as a standalone technique, or continual learning frameworks.

There are 44 multi-task learning frameworks tracked. 4 score above 50 (established tier). The highest-rated is qdrant/quaterion at 62/100 with 661 stars and 93 monthly downloads.

Get all 44 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=multi-task-learning&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

# Framework Score Tier
1 qdrant/quaterion

Blazing fast framework for fine-tuning similarity learning models

62
Established
2 median-research-group/LibMTL

A PyTorch Library for Multi-Task Learning

61
Established
3 zeiss-microscopy/OAD

Collection of tools and scripts useful to automate microscopy workflows in...

58
Established
4 lucidrains/PoPE-pytorch

Efficient implementation (and explorations) into polar coordinate positional...

55
Established
5 Zehong-Wang/Awesome-Weight-Space-Learning

A collection of weight space learning including papers, codes, and datasets.

49
Emerging
6 MR-HosseinzadehTaher/BenchmarkTransferLearning

Official PyTorch Implementation and Pre-trained Models for Benchmarking...

48
Emerging
7 snimu/rebasin

Apply methods described in "Git Re-basin"-paper [1] to arbitrary models ---...

48
Emerging
8 edouardelasalles/stnn

Code for the paper "Spatio-Temporal Neural Networks for Space-Time Series...

47
Emerging
9 Mikoto10032/AutomaticWeightedLoss

Multi-task learning using uncertainty to weigh losses for scene geometry and...

46
Emerging
10 thuml/awesome-multi-task-learning

A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning...

45
Emerging
11 mbs0221/Multitask-Learning

Awesome Multitask Learning Resources

43
Emerging
12 david-knigge/ccnn

Code repository of the paper "Modelling Long Range Dependencies in ND: From...

43
Emerging
13 WeiHongLee/Awesome-Multi-Task-Learning

An up-to-date list of works on Multi-Task Learning

42
Emerging
14 TatsuyaShirakawa/poincare-embedding

Poincaré Embedding (unofficial)

42
Emerging
15 samuela/git-re-basin

Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"

41
Emerging
16 jianghaojun/Awesome-Parameter-Efficient-Transfer-Learning

A collection of parameter-efficient transfer learning papers focusing on...

39
Emerging
17 SimonVandenhende/Awesome-Multi-Task-Learning

A list of multi-task learning papers and projects.

37
Emerging
18 mit-gfx/ContinuousParetoMTL

[ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning

37
Emerging
19 Rishit-dagli/GLOM-TensorFlow

An attempt at the implementation of GLOM, Geoffrey Hinton's paper for...

35
Emerging
20 berkanlafci/oadat

OADAT: Experimental and Synthetic Clinical Optoacoustic Data for...

35
Emerging
21 ryusudol/Centered-Kernel-Alignment

The Fastest, Memory-efficient Python Library for layer-wise similarity...

34
Emerging
22 synbol/Awesome-Parameter-Efficient-Transfer-Learning

Collection of awesome parameter-efficient fine-tuning resources.

34
Emerging
23 joelowj/mtl-tsmom

Multi Task Learning Time Series Momentum

33
Emerging
24 BioDT/bfm-model

Multi-modal Foundation Model for Biodiversity dynamics forecasting

32
Emerging
25 HSG-AIML/SANE

Code Repository for the ICML 2024 paper: "Towards Scalable and Versatile...

31
Emerging
26 AnkurMali/ContinualPTNCN

We introduce Local recurrent Predictive coding model termed as Parallel...

28
Experimental
27 ssnl/poisson_quasimetric_embedding

Open source code for paper "On the Learning and Learnability of Quasimetrics".

27
Experimental
28 BioDT/bfm-data

BioCube: A Multimodal Dataset for Biodiversity

27
Experimental
29 Gxinhu/MTL_airfoil_surrogate

Official implementation for the paper: "Enhancing Airfoil Design...

27
Experimental
30 yashkc2025/low_capacity_nn_behavior

Code for paper "Understanding Generalization, Robustness, and...

27
Experimental
31 ovcharenkoo/mtl_low

Multi-task learning for low-frequency extrapolation and elastic model building

26
Experimental
32 bornabr/CAPC

Context Aware Predictive Coding (CAPC)

23
Experimental
33 deargen/MT-ENet

Repository for "Improving evidential deep learning via multi-task learning,"...

23
Experimental
34 insaneI5/information-sphere

📊 Analyze and assess information structures with the Information Particle...

22
Experimental
35 MinaGhadimiAtigh/Hyperbolic-Busemann-Learning

Hyperbolic Busemann Learning with Ideal Prototypes, NeurIPS2021

21
Experimental
36 HSG-AIML/NeurIPS_2022-Generative_Hyper_Representations

Code Repository for the NeurIPS 2022 paper: "Hyper-Representations as...

20
Experimental
37 Pengxin-Guo/Awesome-Multi-Task-Learning

Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)

19
Experimental
38 YeonwooSung/GLOM

PyTorch implementation of GLOM

19
Experimental
39 HSG-AIML/NeurIPS_2021-Weight_Space_Learning

Code Repository for the NeurIPS 2021 paper: "Self-Supervised Representation...

18
Experimental
40 changsheng137/information-sphere

🌐 Information-Oriented Sphere System: From Data to Information Paradigm...

17
Experimental
41 sdsds222/WarpPCHIP-Net

WarpPCHIP-Net integrates RNN memory, convolutional extraction, and PCHIP...

16
Experimental
42 sdsds222/Simulated-Smooth-RNN

recurrent memory cell for sequence processing. Its core innovation provides...

15
Experimental
43 glassroom/goom_ssm_rnn

Reference implementation of the deep RNN described in "Generalized Orders of...

13
Experimental
44 davidandym/Multi-Task-Optimization

A (hopefully) relatively straightforward, easy to modify code base for...

10
Experimental