Senior Machine Learning Engineer, Data and ML Innovation

What you'll do

• Design and implement large-scale LLM pretraining and post-training pipelines including SFT and preference optimization.
• Drive model hillclimbing through dataset curation, hyperparameter tuning, and ablation studies.
• Develop evaluation frameworks for offline benchmarks and online metrics covering reasoning and tool use.
• Design and maintain verifiers and rubric-based reward systems for model alignment and agentic tasks.
• Build data pipelines for data generation, filtering, labeling, and replay buffers.
• Build and maintain agent training environments including tool APIs, simulators, and sandboxed runtimes.
• Implement environment abstractions to support reinforcement learning and agent evaluation at scale.
• Work on scalable training workflows using distributed frameworks and containerized environments.

What we're looking for

• 5+ years of hands-on ML engineering experience.
• At least 1+ years of experience working directly on large language models or generative AI.
• Bachelor’s, Master’s, or PhD in Computer Science, Machine Learning, or a related technical field, or equivalent practical experience.
• Hands-on experience with LLM training workflows such as pretraining, SFT, RLHF, DPO, or PPO.
• Strong software engineering fundamentals including debugging, testing, code reviews, and production reliability.
• Demonstrated publication records in relevant conferences like NeurIPS, ICML, or ICLR.
• Direct experience with agentic systems, tool use, environment design, or reinforcement learning (preferred).
• Experience with building or operating training environments, simulators, or sandboxed environments (preferred).