Taro Logo

Software Engineer, AI Infrastructure

A leading fintech company democratizing finance for all by making financial markets accessible to everyone.
$157,000 - $185,000
Machine Learning
Mid-Level Software Engineer
In-Person
1,000 - 5,000 Employees
2+ years of experience
AI · Finance

Description For Software Engineer, AI Infrastructure

Robinhood Markets is revolutionizing finance by making financial markets accessible to everyone. As a Software Engineer on the AI Infrastructure team, you'll be at the forefront of building the foundation for ML and Generative AI applications across the company. The role focuses on developing and maintaining a robust, centralized AI platform that enables teams to rapidly build, deploy, and iterate on high-quality Machine Learning applications at scale.

You'll work on critical systems for model deployment, monitoring, and management while collaborating closely with Data Scientists and Applied ML Engineers. Your responsibilities will include designing scalable systems, streamlining ML workflows, developing feature stores, implementing observability solutions, and optimizing cloud resources for training and inference.

The ideal candidate brings 2+ years of software engineering experience, particularly in ML infrastructure or data engineering. You should be proficient in Python or C++, familiar with ML frameworks, and have hands-on experience with modern ML infrastructure tools. The role offers competitive compensation, comprehensive benefits, and the opportunity to work in a collaborative environment with catered meals and well-stocked facilities.

This position is based in either Bellevue, WA or Menlo Park, CA, with a compensation range of $157,000-$185,000 USD, plus bonus and equity opportunities. You'll be part of a mission-driven team focused on democratizing finance while building foundational AI systems that drive business impact across Robinhood.

Last updated 23 days ago

Responsibilities For Software Engineer, AI Infrastructure

  • 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
  • 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 For Software Engineer, AI Infrastructure

Python
Kubernetes
  • 2+ 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
  • A collaborative mindset with an interest in building reliable platforms that support company-wide impact

Benefits For Software Engineer, AI Infrastructure

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
Parental Leave
Commuter Benefits
  • Market competitive and pay equity-focused compensation structure
  • 100% paid health insurance for employees with 90% coverage for dependents
  • Annual lifestyle wallet for personal wellness, learning and development
  • Lifetime maximum benefit for family forming and fertility benefits
  • Dedicated mental health support for employees and eligible dependents
  • Generous time away including company holidays, paid time off, sick time, parental leave
  • Lively office environment with catered meals, fully stocked kitchens, and geo-specific commuter benefits

Interested in this job?

Jobs Related To Robinhood Software Engineer, AI Infrastructure