Add support for multiproblem to mtf_model
Add support for multiproblem to mtf_model

PiperOrigin-RevId: 224051889

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor