search

HomebrewNLP / Olmax

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

Dense2 #87

Closed ClashLuke closed 1 year ago

ClashLuke commented 1 year ago

Just more efficiency for bottleneck block

ClashLuke commented 1 year ago

grafik