Imported from https://github.com/ayrna/deep-ordinal-clm