Add loss like Rho-1#260

GeorgiosSmyrnis wants to merge 4 commits into

mainmlfoundations/open_lm:mainfrom

gsmyrnis/rho1mlfoundations/open_lm:gsmyrnis/rho1Copy head branch name to clipboard