Multi Task Learning Python Packages

bert-multitask-learning

BERT for Multitask Learning

torchjd

Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).

bert-classifier

A repository for training transformer based models

nlp-models

A repository for training transformer based models

eqtransformer

EQTransformer, a python package for earthquake signal detection and phase picking using AI.

phonlp

PhoNLP: A BERT-based multi-task learning model for part-of-speech tagging, named entity recognition and dependency parsing (NAACL 2021)

mmsa

MMSA is a unified framework for Multimodal Sentiment Analysis.

easy-cs-rec-custommodel

A framework for large scale recommendation algorithms.

emmental

A deep learning framework for building multimodal multi-task learning systems.

tasksource

Preprocessings to prepare datasets for a task

ice-learn

A high-level Deep Learning framework that extends PyTorch and PyCUDA.

rec-pangu

rec_pangu is a flexible open-source project for recommendation systems. It incorporates diverse AI models like ranking algorithms, sequence recall, multi-interest models, and graph-based techniques. Designed for both beginners and advanced users, it enables rapid construction of efficient, custom recommendation engines.

pai-easy-rec

A framework for large scale recommendation algorithms.

paddlepalm

a Fast, Flexible, Extensible and Easy-to-use NLP Large-scale Pretraining and Multi-task Learning Framework.

libmtl

A PyTorch Library for Multi-Task Learning

min2net

MIN2Net: End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification

mad-fw

Multitask Aggression Detection (MAD)

toploss

Topological Loss Engineering: differentiable, optimizer-free regularizers that embed 2024-2026 optimizer breakthroughs (Muon/XSAM/CWD/AdEMAMix/NTKMTL/SymNoise) directly into the loss.

mtrec

A simple package about multi-task recommendation

multi-task-nmt-lawhy

The Pytorch implementation of the Multi-task Neural Machine Transliteration system.

m3tl

BERT for Multitask Learning