Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning", augmented with hierarchical task projection - View it on GitHub