SpaceX has almost finished writing v1.0 of an in-house AI training stack in C
entitySpaceX AI Training StackSpaceX AI Training Stack presents a believed speed improvement for large training runs compared to JAX, focusing on high-demand AI training scenarios through optimized architecture using pipeline parallelism to achieve performance close to bare metal.