Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO) - View it on GitHub