Cranial-XIX / FAMOLinks
Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)
☆107Updated last year
Alternatives and similar repositories for FAMO
Users that are interested in FAMO are comparing it to the libraries listed below
Sorting:
- Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]☆235Updated 6 months ago
- Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)☆141Updated 2 years ago
- The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].☆140Updated 2 years ago
- Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"☆388Updated 4 years ago
- Paper List for Multi-Task Learning (focus on architectures and optimization for MTL)☆51Updated 2 years ago
- PyTorch implementation of the GradNorm☆116Updated last year
- The official PyTorch implementation of the paper: Xili Dai, Shengbang Tong, et al. "Closed-Loop Data Transcription to an LDR via Minimaxi…☆64Updated 3 years ago
- Code for "Gradient Surgery for Multi-Task Learning"☆342Updated 5 years ago
- ☆72Updated last year
- [ICLR 24 Oral/Outstanding Paper Honorable Mention Award 🎉]☆39Updated last year
- ☆47Updated 3 years ago
- Code release of paper "ForkMerge: Mitigating Negative Transfer in Auxiliary-Task Learning" (NeurIPS 2023)☆17Updated 2 years ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆144Updated 3 years ago
- PyTorch implementation of "From Sparse to Soft Mixtures of Experts"☆67Updated 2 years ago
- [NeurIPS 2023] The PyTorch Implementation of Scheduled (Stable) Weight Decay.☆61Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆90Updated 2 years ago
- A pure PyTorch implementation of kmeans and GMM with distributed clustering.☆50Updated 3 years ago
- ICLR2023 statistics☆59Updated 2 years ago
- Crawl & Visualize ICLR 2023 Data from OpenReview☆86Updated 2 years ago
- ICLR 2023 Paper submission analysis from https://openreview.net/group?id=ICLR.cc/2023/Conference☆107Updated 3 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆233Updated 2 years ago
- Official repository of "Pareto Manifold Learning: Tackling multiple tasks via ensembles of single-task models" [ICML 2023]☆23Updated 11 months ago
- Official Pytorch's implementation of RotoGrad☆80Updated 2 years ago
- ☆46Updated 3 years ago
- A collection of differentiable SVD methods and ICCV21 "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance P…☆79Updated 2 years ago
- ☆73Updated 2 years ago
- Package for working with hypernetworks in PyTorch.☆131Updated 2 years ago
- Mutual Information Neural Estimation in Pytorch☆347Updated last year
- This is a curated list for Information Bottleneck Principle, in memory of Professor Naftali Tishby.☆383Updated last year
- Code for the ICML 2021 paper "Bridging Multi-Task Learning and Meta-Learning: Towards Efficient Training and Effective Adaptation", Haoxi…☆68Updated 4 years ago