Senior Machine Learning Engineer, AI Infra

Job Description

Lead the architecture and end-to-end delivery of scalable systems for deploying, monitoring, and managing machine learning models in production. Drive cross-functional partnerships to streamline workflows and optimize cloud compute resources for high-throughput training and inference.

Requirements: Requires 6+ years of software engineering experience with deep expertise in ML infrastructure, distributed systems, and production ML workflows. Proficiency in Python or C++ and hands-on experience with ML frameworks and cloud-based infrastructure tools is essential.

Key Skills: Machine learning infrastructure, Python, C++, TensorFlow, PyTorch, Distributed systems, Model serving, Feature store, Observability, AWS, Kubeflow, SageMaker, Ray, Vector databases, Data engineering, Model operations

Benefits: Performance driven compensation, Bonus programs, Equity ownership, 401(k) matching, Health insurance, Employee fund, Lifestyle wallet, Life insurance, Disability insurance, Fertility benefits, Mental health benefits, Paid time off, Parental leave, Catered meals