No flatten depth - Githubissues

HomebrewNLP / Olmax

HomebrewNLP in JAX flavour for maintable TPU-Training

BSD 2-Clause "Simplified" License

45 stars 5 forks source link

Closed ClashLuke closed 1 year ago

ClashLuke commented 1 year ago

should add lax.scan for blocks to avoid 5-hour compile times

ClashLuke commented 1 year ago

deprecated in favor of #99