This model was contributed by sgugger.