dhyaneesh/awesome-jax-flax-llms

A collection of open-source large language model (LLM) implementations in JAX & Flax

3 best points · 0 comments · 1 Hacker News post · 24 GitHub stars · View on GitHub

Hacker News discussions

• Show HN: Jax and Flax LLMs – Transformer Implementations Optimized for TPUs (3 points, 0 comments, by Dhyaneesh, 2025-03-21)