Staff Software Engineer, ML

Job Summary

A company is looking for a Staff Software Engineer, ML Acceleration.

Key Responsibilities

• Analyze ML models to identify and resolve performance bottlenecks
• Incorporate OSS tools for ML engineers to profile and optimize models independently
• Deliver solutions for streamlined model deployment across various hardware platforms

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Engineering, or a related field
• 5+ years of experience, including GPU programming and optimization
• Strong programming skills in C++ and Python
• Proven experience with deep learning frameworks, particularly PyTorch
• Familiarity with CUDA programming and performance optimization techniques