A PyTorch Library for Multi-Task Learning
☆2,583May 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.☆840Mar 3, 2026Updated 5 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]☆246Jun 25, 2025Updated last year
- PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).☆842Jan 13, 2022Updated 4 years ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆141Feb 5, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Multi-task learning using uncertainty to weigh losses for scene geometry and semantics, Auxiliary Tasks in Multi-task Learning☆649Jun 20, 2020Updated 6 years ago
- An up-to-date list of works on Multi-Task Learning☆379Mar 2, 2026Updated 5 months ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆157Nov 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 3 years ago
- The implementation of "End-to-End Multi-Task Learning with Attention" [CVPR 2019].☆734Feb 7, 2022Updated 4 years ago
- LibMOON is a standard and flexible framework to study gradient-based multiobjective optimization.☆142Mar 28, 2025Updated last year
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆403Jun 22, 2021Updated 5 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 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials…☆14,348Feb 18, 2025Updated last year
- MTReclib provides a PyTorch implementation of multi-task recommendation models and common datasets.☆348Jan 23, 2026Updated 6 months ago
- Awesome Multitask Learning Resources☆674Apr 13, 2021Updated 5 years ago
- Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)☆131May 9, 2024Updated 2 years ago
- Implement the Pareto Optimizer and pcgrad to make a self-adaptive loss for multi-task☆44Oct 26, 2021Updated 4 years ago
- 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…☆12,183Mar 16, 2026Updated 4 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆37,063Updated this week
- Code for Neural Information Processing Systems (NeurIPS) 2019 paper: Pareto Multi-Task Learning☆144Jan 2, 2020Updated 6 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆564Nov 1, 2021Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)☆55Nov 30, 2023Updated 2 years ago
- multi task mode for esmm and mmoe☆159Jul 22, 2021Updated 5 years ago
- This repository periodicly updates the MTL paper and resources☆85Aug 5, 2025Updated last year
- All about DeepLearning: 推荐系统、自然语言处理、Tensorflow、Pytorch等☆514May 7, 2025Updated last year
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,951Jul 10, 2026Updated last month
- PyTorch implementation of the GradNorm☆128Sep 4, 2024Updated last year
- [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning☆149Aug 12, 2021Updated 5 years ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆25,480Aug 2, 2026Updated last week
- The project including MMOE, SNR_trans, SNR_avg, PLE, etc implemented by pytorch.☆140Dec 22, 2020Updated 5 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.
- Transfer Learning Library for Domain Adaptation, Task Adaptation, and Domain Generalization☆3,931May 10, 2024Updated 2 years ago
- CVPR 2026 论文和开源项目合集☆22,780Mar 8, 2026Updated 5 months ago
- A playbook for systematically maximizing the performance of deep learning models.☆30,279Jun 18, 2024Updated 2 years ago
- Code of ICLR2023 paper "TaskPrompter: Spatial-Channel Multi-Task Prompting for Dense Scene Understanding" and ECCV2022 paper "Inverted Py…☆326Apr 24, 2024Updated 2 years ago
- Official implementation of Auxiliary learning as an Bargaining Game.☆34Jan 19, 2024Updated 2 years ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆16,036Jul 24, 2024Updated 2 years ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆137Sep 23, 2018Updated 7 years ago