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.

Research

• 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.