Senior/Staff Machine Learning Engineer, Training Runtime Performance

Join the ML Infrastructure team to optimize training runtime efficiency and input pipelines for model training, evaluation, and distillation workloads.
Collaborate with ML practitioners and other infrastructure teams to integrate optimized data pipelines into their workflows.
Detect, diagnose, and resolve performance bottlenecks across training, evaluation, and distillation tasks.
Improve training throughput, resource utilization, and ensure consistent, reproducible model training outcomes.
Enhance input data pipelines to maximize runtime goodput, minimizing idle cycles on accelerators.
Champion best practices for robust, debuggable ML experimentation and contribute to scalable ML systems.