We present SOCKEYE2, a modernized and streamlined version of the SOCKEYE neural machine translation (NMT) toolkit.New features include a simplified code base through the use of MXNet’s GluonAPI, a focus on state-of-the-art model architectures, and distributed mixed precision training. These improvements result in faster training and inference, higher automatic metric scores, and a shorter path from research to production.
Research areas
