A PyTorch Library for Multi-Task Learning
☆2,531May 14, 2025Updated 10 months ago
Alternatives and similar repositories for LibMTL
Users that are interested in LibMTL are comparing it to the libraries listed below
Sorting:
- A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.☆823Mar 3, 2026Updated 2 weeks ago
- Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"☆1,070Sep 2, 2024Updated last year
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆239Jun 25, 2025Updated 8 months ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆840Jan 13, 2022Updated 4 years ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆140Feb 5, 2023Updated 3 years ago
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆647Jun 20, 2020Updated 5 years ago
- An up-to-date list of works on Multi-Task Learning☆377Mar 2, 2026Updated 2 weeks ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆146Nov 9, 2023Updated 2 years ago
- A list of multi-task learning papers and projects.☆385Jan 31, 2022Updated 4 years ago
- A lightweight module for Multi-Task Learning in pytorch.☆168Aug 7, 2023Updated 2 years ago
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆728Feb 7, 2022Updated 4 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆398Jun 22, 2021Updated 4 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆272Aug 9, 2022Updated 3 years ago
- Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials…☆14,298Feb 18, 2025Updated last year
- MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.☆343Jan 23, 2026Updated last month
- Awesome Multitask Learning Resources☆674Apr 13, 2021Updated 4 years ago
- Implement the Pareto Optimizer and pcgrad to make a self-adaptive loss for multi-task☆46Oct 26, 2021Updated 4 years ago
- This repository periodicly updates the MTL paper and resources☆80Aug 5, 2025Updated 7 months ago
- 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…☆12,170Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,504Mar 13, 2026Updated last week
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆566Nov 1, 2021Updated 4 years ago
- Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning☆143Jan 2, 2020Updated 6 years ago
- Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)☆53Nov 30, 2023Updated 2 years ago
- multi task mode for esmm and mmoe☆158Jul 22, 2021Updated 4 years ago
- Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)☆120May 9, 2024Updated last year
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,701Apr 7, 2025Updated 11 months ago
- All about DeepLearning: 推荐系统、自然语言处理、Tensorflow、Pytorch等☆486May 7, 2025Updated 10 months ago
- [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning☆149Aug 12, 2021Updated 4 years ago
- The project including MMOE, SNR_trans, SNR_avg, PLE, etc implemented by pytorch.☆140Dec 22, 2020Updated 5 years ago
- Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization☆3,885May 10, 2024Updated last year
- CVPR 2026 论文和开源项目合集☆22,164Mar 8, 2026Updated last week
- Official implementation of Auxiliary learning as an Bargaining Game.☆32Jan 19, 2024Updated 2 years ago
- A playbook for systematically maximizing the performance of deep learning models.☆29,918Jun 18, 2024Updated last year
- Code of ICLR2023 paper "TaskPrompter: Spatial-Channel Multi-Task Prompting for Dense Scene Understanding" and ECCV2022 paper "Inverted Py…☆327Apr 24, 2024Updated last year
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆15,767Jul 24, 2024Updated last year
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆137Sep 23, 2018Updated 7 years ago
- Awesome Knowledge-Distillation. 分类整理的知识蒸馏paper(2014-2021)。☆2,655May 30, 2023Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,262Oct 17, 2024Updated last year
- Code for "Gradient Surgery for Multi-Task Learning"☆352Apr 28, 2020Updated 5 years ago