Senior Software Engineer - ML Infrastructure

📋 Description

• Design and implement distributed cloud GPU training for DL models
• Build end-to-end ML pipelines and integrate into product workflows
• Encourage ML engineering best practices and maintain high standards
• Collaborate with engineers across the company to solve data problems at scale
• Work across the full ML lifecycle: dataset generation, training, evaluation, deployment

🎯 Requirements

• Bachelor's degree in Computer Science, Software Engineering, or equivalent
• 3+ years of professional experience
• Experience building software components for production ML challenges
• Opinions on building a company-wide ML training/evaluation platform
• Knowledge of OSS landscape and when to use open source vs in-house
• Excellent analytical and problem-solving skills
• Experience with orchestration systems like Airflow and Flyte
• Experience with PyTorch and TensorFlow, and model serving (TorchServe, TensorFlow Serving, Triton)

🎁 Benefits

• Health, dental, vision, life and disability insurance
• 401k with employer match
• Equity in options or RSUs
• Learning and wellness stipends
• Paid time off