消息称xAI拟弃用JAX堆栈，转向自研训练框架

OmniTools 5月30日消息，据SemiAnalysis披露，xAI正逐步放弃基于JAX与NVIDIA GPU的训练堆栈，转而开发自研训练框架以支持Grok系列模型。据报道，xAI现有JAX堆栈的模型算力利用率（MFU）不足10%。此前NVIDIA JAX团队曾长期投入资源支持xAI，目前该合作已告一段落。