Senior Machine Learning Engineer, AI Infra

Join Robinhood's AI Infrastructure team as a Senior Machine Learning Engineer. You'll own the architecture and end-to-end delivery of foundational systems that power model development, deployment, and observability across the company. This role sits at the center of how every AI product at Robinhood gets built, scaled, and maintained in production, working closely with ML practitioners, data engineers, and product teams to define robust, scalable AI platform capabilities.

What you'll do

• Lead the architecture and end-to-end delivery of scalable systems for deploying, monitoring, and managing ML models in production
• Own the technical direction for key platform areas including model serving, the feature store, and ML observability infrastructure, from design through long-term reliability
• Drive cross-functional partnerships with ML practitioners, data engineers, and applied AI teams to streamline workflows, reduce friction, and accelerate experimentation
• Evolve and scale the feature store to support efficient, low-latency feature retrieval across real-time and batch use cases
• Define and implement robust observability standards for model performance, data pipelines, and feature freshness across the ML platform
• Manage and optimize cloud compute resources (CPU/GPU) on AWS to support cost-effective, high-throughput training and inference at scale
• Contribute to technical strategy and roadmap discussions, and mentor engineers on the team through design reviews and hands-on guidance

What you bring

• 6+ years of software engineering experience, with meaningful depth in ML infrastructure, data engineering, or model operations
• Demonstrated ability to own and deliver complex platform systems end-to-end, from architecture to production
• 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 such as TensorFlow or PyTorch
• Solid knowledge of modern ML infrastructure tooling (e.g., Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton)
• Hands-on experience with large-scale search systems, including embedding models, vector databases, and distributed retrieval engines using platforms such as Qdrant, ChromaDB, or Elasticsearch with dense vector search capabilities
• Experience influencing technical direction across teams and mentoring engineers at varying levels
• Bachelor's degree in Computer Science, Software Engineering, or a related technical field

What we offer

• Base pay range (varies by location): Zone 1 (Menlo Park CA, New York NY, Bellevue WA, Washington DC): $209,000-$245,000 USD; Zone 2 (Denver CO, Westlake TX, Chicago IL): $184,000-$216,000 USD; Zone 3 (Lake Mary FL, Clearwater FL, Gainesville FL): $163,000-$191,000 USD
• Bonus opportunities and equity ownership
• 100% paid health insurance for employees with 90% coverage for dependents
• Employer-paid life and disability insurance, fertility benefits, and mental health benefits
• 401(k) matching
• Access to the Robinhood Employee Fund, which gives eligible US employees the opportunity to invest in a private employee fund with exposure to Robinhood Ventures funds
• Lifestyle wallet, a highly flexible benefits spending account for wellness, learning, and more
• Access to leading AI tools on the market and continuous AI skill-building for every employee
• Paid time off, sick time, parental leave, and company holidays
• Exceptional office experience with catered meals, events, and comfortable workspaces

This role is based in Menlo Park, CA or Bellevue, WA with in-person attendance expected at least 3 days per week.

About Robinhood

Robinhood's mission is to democratize finance for all. The company builds products and infrastructure for a historic wealth transfer, with an estimated $124 trillion of assets being inherited by younger generations over the next two decades.