AI-Hypercomputer / maxtextGitHub基于 JAX 的开源大模型库与参考实现,用于在 TPU 和 GPU 上预训练与后训练大语言模型。AI 与机器学习#jax#llm#tpu#pre-training#fine-tuning2444628PythonApache-2.0更新于 4小时前