Distributed Training Engineer optimizes training throughput for video foundation models, enabling researcher experiments with efficient architectures. Requires strong Python skills, distributed systems expertise, and performance profiling.
293k – 490k
HybridML Engineering
About the role
Responsibilities
Collaborate with researchers to enable them to develop systems-efficient video models and architectures
Apply the latest techniques to our internal training framework to achieve impressive hardware efficiency for our training runs
Profile and optimize our training framework
Requirements
Experience working with multi-modal ML pipelines
Love diving deep into systems implementations and understanding their fundamentals in order to improve their performance and maintainability
Strong software engineering skills and are proficient in Python
Experience understanding and optimizing training kernels
Passionate about understanding stable training dynamics
Software Engineer enabling production AI workloads on new hardware platforms through porting, benchmarking, stress testing, and performance optimization. Requires 5+ years in ML systems, distributed training, PyTorch, and RDMA/NCCL expertise.
293k – 455k
Hybrid5+ YOEML Engineering
Research Engineer, Retrieval & Search, Applied Engineering
OpenAISan Francisco, CA
Develop and deploy retrieval and search algorithms for OpenAI's API and ChatGPT, collaborating with research teams on production systems for millions of users. Requires experience with ML systems, vector databases, and large-scale search.
293k – 585k
On-siteML Engineering
Agent Post-Training, API & Power Users
OpenAISan Francisco, CA
Improve agentic model capabilities, reliability, and product fit for power users and API developers through evals, training data, and post-training interventions.
295k – 445k
On-siteML Engineering
Software Engineer, RL Training Infra
OpenAISan Francisco, CA
Build and maintain infrastructure for large-scale RL training runs of frontier OpenAI models. Debug across training, inference, and distributed systems while supporting research integrations.
295k – 445k
HybridML Engineering
Software Engineer, ML Systems & Training Architecture
OpenAISan Francisco, CA
Hands-on senior software engineer focused on maintaining and improving ML training infrastructure, debugging training systems, and unblocking researchers on the robotics team.