Senior Software Engineer, AI Infrastructure (Fintech)

TL;DR

Senior Software Engineer, AI Infrastructure (Fintech): Evolving the ML platform and services to support scalable, reliable, and secure model development and deployment with an accent on accelerating experimentation and improving model observability. Focus on building foundational systems that drive business impact through AI.

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

Salary: $153,000 - $230,000 USD

Company

Robinhood's mission is to democratize finance for all.

What you will do

• Design, build, and maintain scalable systems for deploying, monitoring, and managing Machine Learning models in production.
• Partner with ML practitioners to streamline workflows, integrate internal ML libraries, and optimize model performance.
• Contribute to the development and scaling of our feature store, enabling efficient feature retrieval across real-time and batch use cases.
• Implement robust observability for model performance, data pipelines, and feature freshness.
• Manage and optimize cloud compute resources (CPU/GPU) to support cost-effective training and inference across AWS.

Requirements

• 4+ years of software engineering experience, ideally within ML infrastructure, data engineering, or model operations.
• Hands-on experience with model serving, distributed systems, or production ML workflows.
• Familiarity with modern ML infrastructure tools (e.g., Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton).
• Strong proficiency in Python, C++, or similar languages, and experience with ML frameworks like TensorFlow or PyTorch.
• Hands-on experience with large-scale search systems, including embedding models, vector databases, and distributed retrieval engines, using platforms like Qdrant, ChromaDB, Elasticsearch etc. with dense vector search capabilities.
• A collaborative mindset with an interest in building reliable platforms that support company-wide impact.

Culture & Benefits

• Challenging, high-impact work to grow your career.
• Performance driven compensation with multipliers for outsized impact, bonus programs, equity ownership, and 401(k) matching.
• Best in class benefits to fuel your work, including 100% paid health insurance for employees with 90% coverage for dependents.
• Lifestyle wallet - a highly flexible benefits spending account for wellness, learning, and more.
• Employer-paid life & disability insurance, fertility benefits, and mental health benefits.
• Time off to recharge including company holidays, paid time off, sick time, parental leave, and more!