Senior Software Engineer - ML Infrastructure

We are looking for a Senior Software Engineer - ML Infrastructure to design and implement distributed cloud GPU training approaches for deep learning model training and evaluation. The ideal candidate will have experience with building software components to address production, full-stack machine learning challenges, and knowledge of the open source landscape. Responsibilities include building end-to-end machine learning pipelines, collaborating with engineers across the company, and maintaining a high standard of excellence.

Requirements

• Experience with building software components to address production, full-stack machine learning challenges
• 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
• Excellent analytical and problem-solving skills
• Experience with developing, running, and managing orchestration systems like Airflow and Flyte
• Experience with ML modeling frameworks (PyTorch, Tensorflow, etc.) and model serving platforms (TorchServe, TensorFlow Serving, NVIDIA Triton inference server, etc.)

Benefits

• Comprehensive health, dental, vision, life and disability insurance coverage
• 401k retirement benefits with employer match
• Learning and wellness stipends
• Paid time off