Before he helped build xAI, Jimmy Ba helped solve two quieter problems inside modern AI: how a network takes its next step, and how it stays steady along the way.