Research Engineer, Infrastructure, Numerics

The mission of Thinking Machines is to build AI that extends human will and judgment. We’re looking for an infrastructure research engineer to design and build the core systems that enable efficient large-scale model training with a focus on numerics. You will focus on improving the numerical foundations of our distributed training stack, from precision formats and kernel optimizations to communication frameworks that make training trillion-parameter models stable, scalable, and fast. This role…