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

Star on GitHub
jax-ml

jax-llm-examples

jax-ml/jax-llm-examples

Minimal yet performant LLM examples in pure JAX

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