WeiChengTseng / Pytorch-PCGrad Star 367 Code Issues Pull requests Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning" reinforcement-learning deep-learning deep-reinforcement-learning pytorch mnist rl reimplementation multi-task-learning cifar-100 multi-task-reinforcement-learning multi-task-rl pytorch-pcgrad gradient-surgery mulit-mnist Updated Jun 22, 2021 Python
nslyubaykin / mbrl_multitasking Star 2 Code Issues Pull requests Model-Based RL Multi-Tasking with ReLAx reinforcement-learning reinforcement-learning-algorithms cem continuous-control model-based-reinforcement-learning cross-entropy-method multi-task-reinforcement-learning multi-task-rl Updated Aug 29, 2022 Jupyter Notebook
NoteDance / Pytorch-PCGrad Star 1 Code Issues Pull requests PyTorch Implementation of "Gradient Surgery for Multi-Task Learning" using multiprocessing reinforcement-learning deep-learning deep-reinforcement-learning optimizer pytorch multi-task-learning multi-task-reinforcement-learning multi-task-rl Updated May 17, 2025 Python