Applied ML Engineer—Production Diffusion Pipelines
Listed on 2026-11-05
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Drafted in San Francisco is seeking an ML engineer to help build core infrastructure for training and running our diffusion-model based systems in production. You will collaborate with engineering and research teams to deliver reliable, scalable ML systems.
The role emphasizes production-grade PyTorch code, multi-node training, and potential work with Triton/CUDA kernels to optimize performance. This is a hands-on, in-person position in SF with a small, driven team.
The Applied ML Engineer—Production Diffusion Pipelines role at Drafted is now open for applications in San Francisco, CA, United States.
Learn more about the Applied ML Engineer—Production Diffusion Pipelines role in the description above.
We appreciate your interest in this position.
Join Drafted and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).