Software Engineer, AI Infrastructure

Join a leading fintech company that’s democratizing finance for all.

Robinhood Markets was founded on a simple idea: that our financial markets should be accessible to all. With customers at the heart of our decisions, Robinhood and its subsidiaries and affiliates are lowering barriers and providing greater access to financial information. Together, we are building products and services that help create a financial system everyone can participate in.

With growth as the top priority...

The business is seeking curious, growth-minded thinkers to help shape our vision, structures and systems; playing a key-role as we launch into our ambitious future. If you’re invigorated by our mission, values, and drive to change the world — we’d love to have you apply.

About the team + role

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.

The role is located in the office location(s) listed on this job description which will align with our in-office working environment. Please connect with your recruiter for more information regarding our in-office philosophy and expectations.

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

What we offer

• 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, and more!
• 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, and more!
• Lively office environment with catered meals, fully stocked kitchens, and geo-specific commuter 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 salary range for this role is based on the location where the work will be performed and is aligned to one of 3 compensation zones. This role is also eligible to participate in a Robinhood bonus plan and Robinhood’s equity plan. For other locations not listed, compensation can be discussed with your recruiter during the interview process.

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 here to learn more about available Benefits, which vary by region and Robinhood entity.

We’re looking for more growth-minded and collaborative people to be a part of our journey in democratizing finance for all. If you’re ready to give 100% in helping us achieve our mission—we’d love to have you apply even if you feel unsure about whether you meet every single requirement in this posting. At Robinhood, we're looking for people invigorated by our mission, values, and drive to change the world, not just those who simply check off all the boxes.

Robinhood embraces a diversity of backgrounds and experiences and provides equal opportunity for all applicants and employees. We are dedicated to building a company that represents a variety of backgrounds, perspectives, and skills. We believe that the more inclusive we are, the better our work (and work environment) will be for everyone. Additionally, Robinhood provides reasonable accommodations for candidates on request and respects applicants' privacy rights. Please review the specific Robinhood Privacy Policy applicable to the country where you are applying.