search

HomebrewNLP / Olmax

HomebrewNLP in JAX flavour for maintable TPU-Training
BSD 2-Clause "Simplified" License
45 stars 5 forks source link

Arch2 #105

Open ClashLuke opened 1 year ago

ClashLuke commented 1 year ago

All tests pass, including the leak test. Will have to run the model on a v3-32 to get comparable results.