← all repositories

median-research-group/LibMTL

A PyTorch library implementing multi-task learning algorithms including MMoE, PLE, and gradient-based multi-objective optimization.

2.6k stars Python ML FrameworksLearning
LibMTL
Velocity · 7d
+1.6
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.