Unknown Company

ML Systems Engineer: Trainium Inference & Kernels

san francisco, ca • Posted 1 weeks ago
Onsite Full Time Management & Operations

OpenAI in San Francisco seeks an experienced Software Engineer to help bring inference workloads to AWS Trainium and build the software stack to run frontier models efficiently on the platform. This deeply technical, cross‑stack role covers kernels, compilers, and model execution.

You will develop high‑performance kernels, improve compiler support, and ensure scalable execution of the model forward pass on Trainium.

#J-18808-Ljbffr

ML Systems Engineer: Trainium Inference & Kernels in san francisco at Unknown Company

This position is listed as full time and onsite.

Back to Job Search