A PyTorch Library for Multi-Task Learning
☆2,558May 14, 2025Updated last year
Alternatives and similar repositories for LibMTL
Users that are interested in LibMTL are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A curated list of DATASETS, CODEBASES and PAPERS on Multi-Task Learning (MTL), from Machine Learning perspective.☆837Mar 3, 2026Updated 3 months ago
- Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"☆1,071Sep 2, 2024Updated last year
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆245Jun 25, 2025Updated 11 months ago
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆841Jan 13, 2022Updated 4 years ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆141Feb 5, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆648Jun 20, 2020Updated 5 years ago
- An up-to-date list of works on Multi-Task Learning☆378Mar 2, 2026Updated 3 months ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆152Nov 9, 2023Updated 2 years ago
- A list of multi-task learning papers and projects.☆386Jan 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].☆731Feb 7, 2022Updated 4 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆401Jun 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…☆273Aug 9, 2022Updated 3 years ago
- Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials…☆14,338Feb 18, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.☆347Jan 23, 2026Updated 4 months ago
- Awesome Multitask Learning Resources☆671Apr 13, 2021Updated 5 years ago
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆141Mar 28, 2025Updated last year
- Implement the Pareto Optimizer and pcgrad to make a self-adaptive loss for multi-task☆44Oct 26, 2021Updated 4 years ago
- This repository periodicly updates the MTL paper and resources☆84Aug 5, 2025Updated 10 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,868Jun 3, 2026Updated last week
- 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…☆12,177Mar 16, 2026Updated 2 months ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆567Nov 1, 2021Updated 4 years ago
- Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning☆143Jan 2, 2020Updated 6 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 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☆159Jul 22, 2021Updated 4 years ago
- Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)☆129May 9, 2024Updated 2 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,849May 21, 2026Updated 3 weeks ago
- All about DeepLearning: 推荐系统、自然语言处理、Tensorflow、Pytorch等☆507May 7, 2025Updated last year
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆25,297May 29, 2026Updated 2 weeks 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,913May 10, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- CVPR 2026 论文和开源项目合集☆22,677Mar 8, 2026Updated 3 months ago
- 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.☆30,183Jun 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 2 years ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆15,964Jul 24, 2024Updated last year
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆136Sep 23, 2018Updated 7 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,259Oct 17, 2024Updated last year