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

Star on GitHub
jax-ml

jax-llm-examples

jax-ml/jax-llm-examples

Minimal yet performant LLM examples in pure JAX

247stars
View on GitHub
Forks
34
Open issues
8
Watchers
247
Size
2.8 MB
PythonApache License 2.0
jaxllmllm-inference
Created: Feb 6, 2025
Updated: Apr 13, 2026
Last push: Apr 10, 2026