Star AlbumentationsX on GitHub — it powers this leaderboard

Star on GitHub
← Back to leaderboard
jax-ml

jax-llm-examples

jax-ml/jax-llm-examples

Minimal yet performant LLM examples in pure JAX

240stars
View on GitHub
Forks
31
Open issues
8
Watchers
240
Size
2.8 MB
PythonApache License 2.0
jaxllmllm-inference
Created: Feb 6, 2025
Updated: Feb 24, 2026
Last push: Jan 14, 2026