Machine Learning Engineer

Job Summary

Focusing on optimizing high-performance systems for machine learning, the full-time Machine Learning Systems Engineer will enhance distributed training pipelines, develop GPU kernels, and improve data loading processes, with the flexibility to work remotely or in-office at locations in Boston, Pittsburgh, or Las Vegas.

Key responsibilities

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

Required qualifications

• Bachelor's, Master's degree, or PhD in Computer Science, Computer Engineering, or a related technical 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 skills with a data-driven approach to problem-solving