search

HomebrewNLP / Olmax

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

Flatten Conv for Shampoo, FP64 inverse #96

Closed ClashLuke closed 1 year ago

ClashLuke commented 1 year ago

Closed in favor of #97