Machine Learning Engineer

About the Role

As a Member of Technical Staff, Research Engineer (Model Training), you will play a key role in enhancing AI model performance through sophisticated post-training techniques. Your focus will be on optimizing models beyond initial training, developing advanced fine-tuning methodologies that improve robustness, alignment, and reliability at an enterprise scale.

This Role is Ideal for You If You:

• Have significant experience training large-scale language models and designing complex post-training pipelines.
• Are proficient in deep learning frameworks—especially PyTorch—and have a strong grasp of transformer architectures.
• Have expertise in fine-tuning strategies, including reinforcement learning-based methods like RLHF and DPO.
• Enjoy tackling high-impact challenges and developing novel techniques to improve AI model reliability and alignment.
• Thrive in a collaborative, multidisciplinary environment focused on enterprise success.

Key Responsibilities:

• Design and implement scalable post-training pipelines utilizing advanced fine-tuning methodologies.
• Develop and refine techniques, including reinforcement learning-based approaches, to enhance model alignment, safety, and robustness.
• Collaborate with cross-functional teams to transition research advancements into production-ready AI systems.
• Conduct rigorous experimentation and analysis to continuously improve model performance.
• Serve as a technical leader, driving innovation in AI model training to push the boundaries of enterprise AI.