Pytorch implementation of Adaptive Information Transfer Multi-task (AITM) - View it on GitHub