⭐ Star AlbumentationsX on GitHub — 448+ stars and counting!

Star on GitHub
jax-ml

jax-llm-examples

jax-ml/jax-llm-examples

Minimal yet performant LLM examples in pure JAX

257stars
View on GitHub
Forks
36
Open issues
9
Watchers
257
Size
2.8 MB
PythonApache License 2.0
jaxllmllm-inference
Created: Feb 6, 2025
Updated: May 17, 2026
Last push: Apr 10, 2026