mlfoundations / open_lm

A repository for research on medium sized language models.
MIT License
321 stars 41 forks source link

Add xformers vs torch test #252

Closed achalddave closed 1 month ago

GeorgiosSmyrnis commented 1 month ago

Looks good, I think the only other test that would be good is the nquery < nkeys test as you mentioned offline.