jax-ml/jax-llm-examples
Minimal yet performant LLM examples in pure JAX
46
/ 100
Emerging
244 stars.
No Package
No Dependents
Maintenance
10 / 25
Adoption
10 / 25
Maturity
9 / 25
Community
17 / 25
Stars
244
Forks
32
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 14, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/transformers/jax-ml/jax-llm-examples"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
OptimalScale/LMFlow
An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.
59
adithya-s-k/AI-Engineering.academy
Mastering Applied AI, One Concept at a Time
50
JIA-Lab-research/LongLoRA
Code and documents of LongLoRA and LongAlpaca (ICLR 2024 Oral)
46
riyanshibohra/TuneKit
Upload your data → Get a fine-tuned SLM. Free.
45
young-geng/scalax
A simple library for scaling up JAX programs
45