Senior Software Engineer - ML Infrastructure

Job summary

Responsibilities

We are looking for both infrastructure engineers with expertise in machine learning pipelines and ML engineers that want to work beyond modeling to join the Data & ML infra group. This role will work across the entire ML lifecycle (dataset generation, training frameworks, compute, evaluation, and…

• Design and implement distributed cloud GPU training approaches for deep learning model training and evaluation
• Build end-to-end machine learning pipelines and integrate them into core product workflows
• Encourage change, especially in support of ML engineering best practices, and maintain a high standard of excellence
• Collaborate with engineers across the entire company to solve complex data problems at scale

What they're looking for

• A Bachelor's degree in Computer Science, Software Engineering, or equivalent
• 3+ years of professional experience
• Experience with building software components to address production, full-stack machine learning challenges. This is not purely a research problem
• Opinions about building a company-wide platform for ML training, evaluation, and deployment
• Knowledge of the open source landscape with judgment on when to choose open source versus build in-house

Tech stack

• Airflow
• PyTorch
• TensorFlow
• Triton