Software Engineer, ML Infrastructure
Serve Robotics119,000 - 160,000 USD/year
- Full Time
- Junior (1 to 2 years)
$295,000 – $440,000Compensation
Mid-level (3 to 4 years), Senior (5 to 8 years)Experience Level
Full TimeJob Type
UnknownVisa
AI & Machine Learning, Data & Analytics, Enterprise SoftwareIndustries
Requirements
Candidates should have a strong technical background in data technologies, distributed systems, and reliable software development. Deep expertise in ML system optimization, distributed systems, or full-stack application development for internal tools is required. Experience collaborating with ML researchers in an applied setting is highly valued. Proficiency in debugging ML systems, experience with reinforcement learning and transformers, and familiarity with Python, Kubernetes, distributed infrastructure, GPUs, and large-scale data systems such as Beam or Spark are essential.
Responsibilities
The Research Infrastructure Engineer will ensure that systems powering ChatGPT training and development run smoothly. They will dive into large ML codebases to understand and debug system issues, work with researchers to build tools for data management, model configuration, and evaluation, and create reusable Python libraries for ML projects. Additional responsibilities include profiling large model reinforcement learning training, identifying experiment failures in new research clusters, redesigning data pipelines for diverse multimodal data, and building front-end evaluation tooling for company-wide use.
Skills
Python
Kubernetes
GPUs
Beam
Spark
Reinforcement Learning
Transformers
Distributed Systems
Data Technologies
Debugging ML Systems
OpenAI
Develops safe and beneficial AI technologies
About OpenAI
OpenAI develops and deploys artificial intelligence technologies aimed at benefiting humanity. The company creates advanced AI models capable of performing various tasks, such as automating processes and enhancing creativity. OpenAI's products, like Sora, allow users to generate videos from text descriptions, showcasing the versatility of its AI applications. Unlike many competitors, OpenAI operates under a capped profit model, which limits the profits it can make and ensures that excess earnings are redistributed to maximize the social benefits of AI. This commitment to safety and ethical considerations is central to its mission of ensuring that artificial general intelligence (AGI) serves all of humanity.
Key Metrics
San Francisco, CaliforniaHeadquarters
2015Year Founded
$18,433.2MTotal Funding
LATE_VCCompany Stage
AI & Machine LearningIndustries
1,001-5,000Employees