Machine Learning Engineer

Job Summary

A company is looking for a Member of Technical Staff, Machine Learning Optimization.

Key Responsibilities

• Improve performance of model training and inference
• Optimize state-of-the-art AI models for video generation
• Build tooling to enhance efficiency and reliability of distributed training runs

Required Qualifications

• 5 years of experience in a similar software engineering position
• 5 years of experience optimizing machine learning model inference and training
• 5 years of experience with Python, C/C++, and CUDA
• 5 years of experience profiling GPU performance and distributed training runs
• Bachelor's degree in Computer Science, Mathematics, Applied Mathematics, or a closely related field