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 3 hours ago
#pre-training
Curated GitHub repositories related to pre-training, with AI summaries and key stats.
A JAX-based open-source LLM library and reference implementation for pre-training and post-training large models on Google Cloud TPUs and GPUs.