nmi-val / softlearning / models