Senior Machine Learning Engineer, AI Infra

Location: Menlo Park, CA or Bellevue, WA; in-person attendance is expected at least 3 days per week.

Base pay: $163,000–$191,000, $184,000–$216,000, or $209,000–$245,000 USD annually, depending on compensation zone. Additional bonus, equity, and benefits are available.

Company

Robinhood builds financial products and technology aimed at democratizing finance.

What you will do

• Lead the architecture and end-to-end delivery of scalable systems for deploying, monitoring, and managing ML models in production.
• Own technical direction for model serving, the feature store, and ML observability infrastructure.
• Scale the feature store for low-latency retrieval across real-time and batch use cases.
• Define observability standards for model performance, data pipelines, and feature freshness.
• Manage and optimize AWS CPU/GPU resources for high-throughput training and inference.
• Partner with ML practitioners, data engineers, and applied AI teams while mentoring engineers and contributing to technical strategy.

Requirements

• 6+ years of software engineering experience with depth in ML infrastructure, data engineering, or model operations.
• Experience owning complex platform systems from architecture through production.
• Expertise in model serving, distributed systems, and production ML workflows at scale.
• Strong proficiency in Python, C++, or similar languages, with hands-on TensorFlow or PyTorch experience.
• Knowledge of Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton, or comparable ML infrastructure tools.
• Experience with embedding models, vector databases, and distributed retrieval systems such as Qdrant, ChromaDB, or Elasticsearch dense vector search.

Nice to have

• Advanced degree in Computer Science, Software Engineering, or a related technical field.

Culture & Benefits

• High-impact work on financial technology and AI infrastructure.
• Performance-driven compensation with bonuses, equity ownership, and 401(k) matching.
• US employee benefits include paid health insurance, with dependent coverage, and other insurance programs.
• Paid time off, sick time, company holidays, parental leave, and mental health and fertility benefits.
• Access to AI tools, continuous AI skill-building, a flexible lifestyle wallet, and an office experience with catered meals and events.