Machine Learning Engineer

Job Summary

Focused on optimizing large-scale distributed model training, the full-time Machine Learning Systems Engineer will enhance performance through profiling, kernel development, and data pipeline engineering, with opportunities for hybrid or fully remote work.

Key responsibilities

• Utilize profiling tools to identify and resolve bottlenecks in data loading and computation
• Optimize distributed training pipelines using frameworks such as PyTorch Distributed
• Design and maintain high-performance GPU kernels for machine learning workloads

Required qualifications

• Bachelor's, Master's degree, or PhD in Computer Science, Computer Engineering, or a related discipline
• Strong proficiency in Python programming
• Extensive hands-on experience with PyTorch
• Experience in optimizing machine learning model execution during training and inference
• Exceptional analytical and problem-solving skills with a data-driven approach