xAI replaces JAX framework with custom C++ codebase for Grok 4.8
Chips & ComputeThe Neuron · 2h ago

xAI replaces JAX framework with custom C++ codebase for Grok 4.8

Elon Musk announced that xAI transitioned its model training software from Google JAX to a proprietary C and C++ system. The engineering swap comes as Grok 4.8 undergoes reinforcement training ahead of an upcoming soft release.

Elon MuskxAI
Read the original