search

HomebrewNLP / Olmax

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

Allow broken TPUs + Fix inference #41

Closed ClashLuke closed 2 years ago

ClashLuke commented 2 years ago

Unless we can fix typical sampling, this is the only salvage part from #39.