Software Engineer, AI Infrastructure

New Yesterday

Join us in building the future of finance.
Our mission is to democratize finance for all. An estimated $124 trillion of assets will be inherited by younger generations in the next two decades. The largest transfer of wealth in human history. If you’re ready to be at the epicenter of this historic cultural and financial shift, keep reading. About the team + role
We are building an elite team, applying frontier technologies to the world’s biggest financial problems. We’re looking for bold thinkers. Sharp problem-solvers. Builders who are wired to make an impact. Robinhood isn’t a place for complacency, it’s where ambitious people do the best work of their careers. We’re a high-performing, fast-moving team with ethics at the center of everything we do. Expectations are high, and so are the rewards.
The AI Infrastructure team’s mission is to provide a robust, agile, and centralized AI platform—empowering teams across Robinhood to rapidly build, deploy, and iterate on high-quality Machine Learning (ML) and Generative AI (GenAI) applications at scale.
As a Software Engineer, you’ll help evolve our ML platform and services to support scalable, reliable, and secure model development and deployment. You’ll work closely with Data Scientists and Applied ML Engineers to build foundational systems that accelerate experimentation, improve model observability, and drive business impact through AI.
This role is based in our Menlo Park, CA or Bellevue, WA office(s), with in-person attendance expected at least 3 days per week.
At Robinhood, we believe in the power of in-person work to accelerate progress, spark innovation, and strengthen community. Our office experience is intentional, energizing, and designed to fully support high-performing teams.
What you’ll 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. What you bring
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.
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. What we offer
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!
Exceptional office experience with catered meals, events, and comfortable workspaces.
In addition to the base pay range listed below, this role is also eligible for bonus opportunities + equity + benefits.
Base pay for the successful applicant will depend on a variety of job-related factors, which may include education, training, experience, location, business needs, or market demands. The expected base pay range for this role is based on the location where the work will be performed and is aligned to one of 3 compensation zones. For other locations not listed, compensation can be discussed with your recruiter during the interview process.
Base Pay Range: Zone 1 (Menlo Park, CA; New York, NY; Bellevue, WA; Washington, DC)$157,000—$185,000 USDZone 2 (Denver, CO; Westlake, TX; Chicago, IL)$139,000—$163,000 USDZone 3 (Lake Mary, FL; Clearwater, FL; Gainesville, FL)$122,000—$144,000 USD Click to learn more about our Total Rewards, which vary by region and entity.
If our mission energizes you and you’re ready to build the future of finance, we look forward to seeing your application.
Robinhood provides equal opportunity for all applicants, offers reasonable accommodations upon request, and complies with applicable equal employment and privacy laws. Inclusion is built into how we hire and work—welcoming different backgrounds, perspectives, and experiences so everyone can do their best. Please review the for your country of application.
Apply
Location:
Bellevue

We found some similar jobs based on your search