Imported from https://github.com/allenai/bilm-tf