repoGitHubTrust 82 · PrimaryPublished 12d agoLive · 12d ago
jax-ml/jax-llm-examples
Minimal yet performant LLM examples in pure JAX
Lineage graph
Paper → model → repo connections mined from source citations (Tier-1 exact match).
Why these links exist
Every edge carries a method, confidence, and the source snippet that justified it — so bad links are debuggable.
- PossiblePossibly related (embedding) · 56%Reducing High-Bandwidth Memory Bottlenecks in JAX-Based LLM Training with Host Offloading - NVIDIA Developer →
