This project is a re-implementation of MMoE Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts. The reference code is the keras version of MMoE: keras-version
mmoe = MMoEModule(input_size, units, num_experts, num_tasks)
output = mmoe(input)