Imported from https://github.com/nyu-mll/jiant