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 46 multi-task learning frameworks tracked. 4 score above 50 (established tier). The highest-rated is zeiss-microscopy/OAD at 58/100 with 162 stars.

Get all 46 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	Stars	Language
1	zeiss-microscopy/OAD Collection of tools and scripts useful to automate microscopy workflows in...	58	Established	162	Jupyter Notebook
2	median-research-group/LibMTL A PyTorch Library for Multi-Task Learning	57	Established	2,531	Python
3	qdrant/quaterion Blazing fast framework for fine-tuning similarity learning models	57	Established	661	Python
4	lucidrains/PoPE-pytorch Efficient implementation (and explorations) into polar coordinate positional...	55	Established	57	Python
5	MR-HosseinzadehTaher/BenchmarkTransferLearning Official PyTorch Implementation and Pre-trained Models for Benchmarking...	48	Emerging	61	Python
6	edouardelasalles/stnn Code for the paper "Spatio-Temporal Neural Networks for Space-Time Series...	47	Emerging	121	Jupyter Notebook
7	Mikoto10032/AutomaticWeightedLoss Multi-task learning using uncertainty to weigh losses for scene geometry and...	46	Emerging	647	Python
8	Zehong-Wang/Awesome-Weight-Space-Learning A collection of weight space learning including papers, codes, and datasets.	46	Emerging	43	—
9	snimu/rebasin Apply methods described in "Git Re-basin"-paper [1] to arbitrary models ---...	46	Emerging	15	Python
10	thuml/awesome-multi-task-learning A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning...	45	Emerging	822	—
11	mbs0221/Multitask-Learning Awesome Multitask Learning Resources	43	Emerging	674	—
12	david-knigge/ccnn Code repository of the paper "Modelling Long Range Dependencies in ND: From...	43	Emerging	183	Python
13	WeiHongLee/Awesome-Multi-Task-Learning An up-to-date list of works on Multi-Task Learning	42	Emerging	376	—
14	TatsuyaShirakawa/poincare-embedding Poincaré Embedding (unofficial)	42	Emerging	229	C++
15	samuela/git-re-basin Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"	41	Emerging	509	Python
16	joelowj/mtl-tsmom Multi Task Learning Time Series Momentum	40	Emerging	25	Python
17	BioDT/bfm-model Multi-modal Foundation Model for Biodiversity dynamics forecasting	39	Emerging	7	Python
18	jianghaojun/Awesome-Parameter-Efficient-Transfer-Learning A collection of parameter-efficient transfer learning papers focusing on...	39	Emerging	410	—
19	ryusudol/Centered-Kernel-Alignment The Fastest, Memory-efficient Python Library for layer-wise similarity...	38	Emerging	4	Python
20	SimonVandenhende/Awesome-Multi-Task-Learning A list of multi-task learning papers and projects.	37	Emerging	384	—
21	mit-gfx/ContinuousParetoMTL [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning	37	Emerging	149	Jupyter Notebook
22	Rishit-dagli/GLOM-TensorFlow An attempt at the implementation of GLOM, Geoffrey Hinton's paper for...	35	Emerging	37	Python
23	AnkurMali/ContinualPTNCN We introduce Local recurrent Predictive coding model termed as Parallel...	35	Emerging	12	Python
24	berkanlafci/oadat OADAT: Experimental and Synthetic Clinical Optoacoustic Data for...	35	Emerging	35	Python
25	synbol/Awesome-Parameter-Efficient-Transfer-Learning Collection of awesome parameter-efficient fine-tuning resources.	34	Emerging	588	—
26	BioDT/bfm-data BioCube: A Multimodal Dataset for Biodiversity	34	Emerging	5	Jupyter Notebook
27	Gxinhu/MTL_airfoil_surrogate Official implementation for the paper: "Enhancing Airfoil Design...	34	Emerging	5	Python
28	yashkc2025/low_capacity_nn_behavior Code for paper "Understanding Generalization, Robustness, and...	33	Emerging	13	Python
29	HSG-AIML/SANE Code Repository for the ICML 2024 paper: "Towards Scalable and Versatile...	31	Emerging	30	Python
30	bornabr/CAPC Context Aware Predictive Coding (CAPC)	30	Emerging	6	Python
31	deargen/MT-ENet Repository for "Improving evidential deep learning via multi-task learning,"...	30	Emerging	20	Python
32	MinaGhadimiAtigh/Hyperbolic-Busemann-Learning Hyperbolic Busemann Learning with Ideal Prototypes, NeurIPS2021	28	Experimental	25	Python
33	HSG-AIML/NeurIPS_2022-Generative_Hyper_Representations Code Repository for the NeurIPS 2022 paper: "Hyper-Representations as...	27	Experimental	18	Python
34	ssnl/poisson_quasimetric_embedding Open source code for paper "On the Learning and Learnability of Quasimetrics".	27	Experimental	32	C++
35	YeonwooSung/GLOM PyTorch implementation of GLOM	26	Experimental	23	Python
36	ovcharenkoo/mtl_low Multi-task learning for low-frequency extrapolation and elastic model building	26	Experimental	31	Jupyter Notebook
37	HSG-AIML/NeurIPS_2021-Weight_Space_Learning Code Repository for the NeurIPS 2021 paper: "Self-Supervised Representation...	25	Experimental	22	Python
38	insaneI5/information-sphere 📊 Analyze and assess information structures with the Information Particle...	22	Experimental	—	Python
39	changsheng137/information-sphere 🌐 Information-Oriented Sphere System: From Data to Information Paradigm...	21	Experimental	2	Python
40	sdsds222/WarpPCHIP-Net WarpPCHIP-Net integrates RNN memory, convolutional extraction, and PCHIP...	20	Experimental	1	—
41	Pengxin-Guo/Awesome-Multi-Task-Learning Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)	19	Experimental	53	—
42	glassroom/goom_ssm_rnn Reference implementation of the deep RNN described in "Generalized Orders of...	19	Experimental	7	Python
43	davidandym/Multi-Task-Optimization A (hopefully) relatively straightforward, easy to modify code base for...	17	Experimental	1	—
44	sdsds222/Simulated-Smooth-RNN recurrent memory cell for sequence processing. Its core innovation provides...	17	Experimental	—	—
45	eth-siplab/Shift-Invariant_Deep_Learning_on_Time_Series The official implementation of ICLR2025 paper for shift-invariant neural networks	15	Experimental	11	Python
46	rbd079/multiparameter_DOT_dataset Simulated frequency-domain diffuse optical tomography dataset	11	Experimental	3	—

Comparisons in this category

awesome-multi-task-learning and Awesome-Multi-Task-Learning (45 vs 37) awesome-multi-task-learning and Awesome-Multi-Task-Learning (45 vs 42) Multitask-Learning and Awesome-Multi-Task-Learning (43 vs 42)