Home / Engineering / Research Engineer, ML Infrastructure

Research Engineer, ML Infrastructure

Chai Discovery · San Francisco office

Research Engineer, ML Infrastructure is open at Chai Discovery in San Francisco office. The posting gives no salary; similar roles in the United States advertise a median of $194K. Posted on Sep 24, 2026, and still on Chai Discovery’s careers page when we last checked, on Nov 14, 2026.

About the role

Make our models performant, resource efficient and reliable at scale by developing the core frameworks for model training and evaluation, in close partnerships with fellow researchers and engineers. - Build our training stack across model, layer, and kernel levels; optimize workloads through parallelism, quantization, and custom kernels. - Profile end-to-end training runs on large GPU clusters; eliminate bottlenecks…

The opening lines of the posting. The rest is on Chai Discovery’s careers page.

Hiring at Chai Discovery

Chai Discovery has 23 open roles right now. Most are in Engineering, Data & Research and Sales & Business Development. See all its roles

Similar Engineering jobs

All Engineering jobs
Apply on company site