A JAX-based open-source LLM library and reference implementation for pre-training and post-training large models on Google Cloud TPUs and GPUs.
2.4K628PythonApache-2.0Updated 5 hours ago
#tpu
Open-source projects tagged “tpu”
Curated GitHub repositories related to tpu, with AI summaries and key stats.