Tags: #jax
LLM Training Framework
python
2.2k
AI-Hypercomputer/maxtext
A high-performance, scalable JAX-based open-source library for training large language models on Google Cloud TPUs and GPUs.
A high-performance, scalable JAX-based open-source library for training large language models on Google Cloud TPUs and GPUs.