Senior Machine Learning Engineer, AI Infra

📋 Description

• Lead the architecture and end-to-end delivery of scalable systems for deploying, monitoring, and
• Own the technical direction for key platform areas — including model serving, the feature store
• Drive cross-functional partnerships with ML practitioners, data engineers, and applied AI teams to
• Evolve and scale our feature store to support efficient, low-latency feature retrieval across
• Define and implement robust observability standards for model performance, data pipelines, and
• Manage and optimize cloud compute resources (CPU/GPU) on AWS to support cost-effective

🎯 Requirements

• 6+ years of software engineering experience, with meaningful depth in ML infrastructure, data
• Demonstrated ability to own and deliver complex platform systems end-to-end, from architecture to
• Deep expertise in model serving, distributed systems, and production ML workflows at scale
• Strong proficiency in Python, C++, or similar languages, and hands-on experience with ML frameworks
• Solid knowledge of modern ML infrastructure tooling (e.g., Ray, Kubeflow, SageMaker, TensorFlow
• Hands-on experience with large-scale search systems, including embedding models, vector databases

🎁 Benefits

• Challenging, high-impact work to grow your career
• Performance driven compensation with multipliers for outsized impact, bonus programs, equity
• Top Tier benefits to fuel your work, including 100% paid health insurance for employees with 90%
• Access to the Robinhood Employee Fund that gives eligible US employees the opportunity to invest in
• Access to the best AI tools on the market and continuous AI skill-building for every employee
• Lifestyle wallet - a highly flexible benefits spending account for wellness, learning, and more