median-research-group/LibMTL
A PyTorch library implementing multi-task learning algorithms including MMoE, PLE, and gradient-based multi-objective optimization.

LibMTL is an open-source PyTorch library for multi-task learning that provides standardized implementations of various MTL architectures and optimization algorithms. It includes popular models like Multi-Gate Mixture of Experts (MMoE) and Progressive Layered Extraction (PLE), as well as gradient manipulation techniques for multi-objective optimization. The library serves as a research framework and practical tool for developing and comparing multi-task deep learning models.