RecodeX 重构消息,作者从头用JAX构建并训练了GPT-2模型,性能接近OpenAI原版,表明用JAX训练大模型可行。