GitHub repository: HomebrewNLP/Olmax
Primary language: Python
Description: HomebrewNLP in JAX flavour for maintable TPU-Training