Senior Machine Learning Engineer, AI Infra

Robinhood
Menlo Park, Bellevue
Workplace: OnsiteFull timeUSD 163,000 - 245,000 annuallyFunction: Data Science & Machine LearningExperience: 6+ yearsEducation: bachelorsSkills: ["Problem-solving","Mentorship","Cross-functional collaboration","Technical strategy","Ownership"]

Own the architecture and end-to-end delivery of AI platform systems powering model development, deployment, and observability. Lead foundational work across model serving, a scalable feature store, and ML observability, partnering with ML, data engineering, and applied AI teams to streamline workflows and accelerate experimentation. Drive reliability, scalability, and craft while optimizing AWS CPU/GPU compute and mentoring engineers through design reviews and hands-on guidance.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Robinhood
Robinhood
1 day ago

Senior Machine Learning Engineer, AI Infra

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 7 hours agoStatus: Live

Job Summary

Own the architecture and end-to-end delivery of AI platform systems powering model development, deployment, and observability. Lead foundational work across model serving, a scalable feature store, and ML observability, partnering with ML, data engineering, and applied AI teams to streamline workflows and accelerate experimentation. Drive reliability, scalability, and craft while optimizing AWS CPU/GPU compute and mentoring engineers through design reviews and hands-on guidance.
Location: Menlo Park, Bellevue
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Lead 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 to ensure long-term reliability.
  • •Drive cross-functional partnerships with ML practitioners, data engineers, and applied AI teams to streamline workflows and accelerate experimentation.
  • •Evolve and scale the feature store for efficient low-latency feature retrieval for real-time and batch use cases.
  • •Define observability standards for model performance, data pipelines, and feature freshness; manage and optimize AWS CPU/GPU compute resources.

Pay and Benefits

Salary: USD 163,000 - 245,000 annually
Perks:Health Insurance401kEquityAnnual BonusPaid LeaveSick TimeParental LeaveLife InsuranceDisability InsuranceLearning Budget

Key Requirements

  • •6+ years of software engineering experience with depth in ML infrastructure, data engineering, or model operations.
  • •Ability to own and deliver complex platform systems end-to-end, from architecture through production.
  • •Deep expertise in model serving, distributed systems, and production ML workflows at scale.
  • •Strong proficiency in Python, C++, or similar languages, with hands-on experience in ML frameworks such as TensorFlow or PyTorch.
  • •Experience with ML infrastructure tooling (e.g., Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton) and at least one large-scale search/vector retrieval stack.
Experience:6+ years
Education:Bachelor's in Computer Science, Software Engineering, or a related technical field
Skills:Problem-solvingMentorshipCross-functional collaborationTechnical strategyOwnership
Languages:English
Tech Stack:PythonC++TensorFlowPyTorchRayKubeflowSageMakerTensorFlow ServingTritonAWSQdrantChromaDBElasticsearch

Company Brief

Robinhood
Commission-free trading platform offering stocks, ETFs, options, and cryptocurrencies, plus cash management and retirement products. Aims to democratize finance through easy-to-use mobile and web apps for retail investors.
Industry: Trading Platforms
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Menlo Park, United States
Founded: 2013
WebsiteLinkedIn