evanatyourservice/llm-jax

Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.

⭐ 19 ⑂ 3 Python MIT · 2025-07-24推送

flax gpt huggingface jax large-language-models llm optimization fineweb

19

Watchers

0

贡献者

0

Commits

0

Releases

0

Open Issues

2025-07-24

最近推送

原文中文

暂无 README