median-research-group/LibMTL

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

2593 stars · 233 forks observed · 2026-09-21

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

LibMTL is an open-source PyTorch library for Multi-Task Learning (MTL). It provides implementations of many MTL architectures and gradient-based multi-objective optimization methods with a unified API.

Use cases

• train multi-task learning models in pytorch
• compare MTL weighting algorithms like PCGrad and CAGrad
• implement MMoE or PLE architectures
• run multi-objective optimization for deep learning
• benchmark gradient manipulation methods for MTL

When to choose

• you need a unified PyTorch framework for multi-task learning experiments
• you want to evaluate many recent MTL optimization methods without reimplementing them
• you need architectures like MMoE and PLE out of the box

When to avoid

• you need single-task training only
• you work outside PyTorch
• you need production serving rather than research training

Facets

library · maturity active

machine-learning deep-learning machine-learning deep-learning python pytorch multi-task-learning mtl multiobjective-optimization mmoe ple

Member repositories

For agents

markdown · JSON · MCP: product_card(name="median-research-group/LibMTL")