Senior / Staff Machine Learning Engineer, Applied AI
Core
Building production-oriented ML systems for LLM training, evaluation, and adaptation to customer-specific scientific workflows.
Role type
Senior/Staff Machine Learning Engineer (Applied AI)
Builds
Reliable AI workflows, evaluation loops, and model adaptation tooling for scientific discovery.
Domain
AI Research / Machine Learning / Scientific Computing
Deliverable
production ML models
Required skills
LLM training and adaptation, distributed ML frameworks (Megatron-LM, TorchTitan, DeepSpeed, Ray), Python, PyTorch/JAX/TensorFlow, experiment design, model debugging, cross-functional collaboration
Preferred skills
Agentic AI systems, RL post-training (RLHF, GRPO), MoE architectures, retrieval-augmented generation, model monitoring dashboards
Technologies
PyTorch, JAX, TensorFlow, Megatron-LM, TorchTitan, DeepSpeed, Ray
Responsibilities
Close the gap between model capabilities and customer scientific workflows; Build evaluation loops for model quality; Design experiments to improve model performance; Integrate model behavior into end-to-end product workflows; Debug model failures using traces and feedback; Build reusable tooling for model adaptation and deployment
Seniority
Senior/Staff, hands-on IC
