AI Summary
Join Code Metal AI as a Senior AI Engineer to work on pioneering projects in large language models and code generation. Develop robust distributed training systems, design scalable data curation pipelines, and drive innovation through evaluation frameworks and reinforcement learning solutions.
Key Highlights
Distributed training systems with PyTorch
Scalable data curation and quality assurance pipelines
Evaluation frameworks and reinforcement learning solutions
Key Responsibilities
Build and maintain robust distributed training systems using PyTorch
Design and implement scalable data curation and quality assurance pipelines
Develop orchestration tools that manage complex workflows across large-scale AI model training and evaluation
Drive innovation by developing evaluation frameworks and reinforcement learning solutions
Technical Skills Required
Benefits & Perks
Health care plan with 100% premium coverage
401k with 5% matching
Paid Time Off (Uncapped Vacation, plus Sick & Public Holidays)
Flexible hybrid work arrangement
Relocation assistance for qualifying employees
Nice to Have
Contributions to open-source AI or ML projects
Published work or demonstrable research experience in related fields
Hands-on experience applying RLHF to LLMs, especially for code generation
Job Description
At Code Metal AI, you’ll be part of a world class team with talent from MIT, OpenAI and other top companies, focused on pioneering work in large language models (LLMs) and code generation. Our projects directly involve leading chip manufacturers, applying advanced AI to solve meaningful, practical challenges with real world impact.
This Role Bridges Two Critical Areas
Production
- Build and maintain robust distributed training systems using PyTorch (2+ years experience required).
- Design and implement scalable data curation and quality assurance pipelines to ensure top-tier training datasets.
- Develop orchestration tools that manage complex workflows across large-scale AI model training and evaluation.
- Drive innovation by developing evaluation frameworks and reinforcement learning solutions, including recent advancements in Reinforcement Learning with Human Feedback (RLHF).
- Engage with frontier research through open-source projects and potential publications, applying RLHF to Large Language Models (LLMs), ideally focusing on code generation tasks.
Looking to advance your Development & Programming career with relocation support? Explore Development & Programming Jobs with Relocation Packages that include comprehensive packages to help you move and settle in your new role.
- 2+ years experience in distributed training, preferably with PyTorch.
- Strong background in reinforcement learning, with recent RLHF experience highly preferred.
- Proven ability to build data curation and quality assurance pipelines.
- Experience with evaluation framework development.
- Ideally, experience across both data pipeline and orchestration sides.
- Eligible for TS/SCI clearance.
Discover our full range of relocation jobs with comprehensive support packages to help you relocate and settle in your new location.
- Contributions to open-source AI or ML projects.
- Published work or demonstrable research experience in related fields.
- Hands-on experience applying RLHF to LLMs, especially for code generation.
- Experience with large scale synthetic data generation.
- Health care plan with 100% premium coverage, including medical, dental, and vision.
- 401k with 5% matching.
- Paid Time Off (Uncapped Vacation, plus Sick & Public Holidays).
- Flexible hybrid work arrangement.
- Relocation assistance for qualifying employees.
Similar Jobs
Explore other opportunities that match your interests
Technical Account Manager
••••••
••••••
••••••
Job Type
••••••
Experience Level
••••••
presales collective
United State
Visa Sponsorship
Relocation
Remote
Job Type
Full-time
Experience Level
Mid-Senior level
chemanager international
United State
GNC Engineer: Autopilot – Level 4
••••••
••••••
••••••
Job Type
••••••
Experience Level
••••••
Northrop Grumman
United State