Machine Learning Engineer
Develops and trains large-scale diffusion models for image/video generation, controllability modules like IPAdapters/ControlNets, and novel research techniques for production. Requires proven experience with image/video models at scale and deep learning frameworks.
About the job
Responsibilities
- Train foundation diffusion models for image and video generation.
- Train controllability modules such as IPAdapters or ControlNets.
- Develop novel research techniques and put them into production.
- Conduct large-scale experiments on high-performance computing clusters, optimizing data pipelines for massive image datasets.
Requirements
- Proven track record in working with image or video models at scale (publications or open-source contributions a plus).
- Strong background in deep learning frameworks and distributed training paradigms.
- Ability to iterate rapidly, and propose creative research directions.
Skills
Diffusion Models, Ipadapters, Controlnets, Deep Learning, PyTorch, TensorFlow, Distributed Training, Image Generation, Video Generation, High-Performance Computing
Similar jobs
ML Engineering jobs
Anthropic
San Francisco, CABuild and optimize a high-scale LLM inference engine spanning accelerator programming, host-device coordination, and distributed systems. The role requires strong systems programming, performance analysis, and an understanding of LLM inference across compute, memory, and interconnects.
OnePay
United States
Build and operate production AI agents, automation workflows, and integrations that improve complex business processes. The role requires 5+ years of software engineering experience, modern LLM and agent-framework expertise, systems integration skills, and strong cross-functional collaboration.
Thinking Machines Lab
San Francisco, CA
Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.
OpenAI
San Francisco, CA
Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.