Software Engineer, AI Infrastructure
Description For Software Engineer, AI Infrastructure
Robinhood Markets, a pioneering fintech company, is seeking a Software Engineer for their AI Infrastructure team. This role is crucial in their mission to democratize finance for all by making financial markets accessible to everyone. The position offers an exciting opportunity to work at the intersection of finance and artificial intelligence, building the foundational systems that power Robinhood's ML and GenAI applications.
The role involves designing and maintaining scalable systems for ML model deployment, working closely with data scientists and ML engineers to optimize workflows and improve model observability. You'll be responsible for developing the feature store, implementing robust monitoring systems, and managing cloud computing resources efficiently across AWS infrastructure.
The ideal candidate should bring 2+ years of software engineering experience, particularly in ML infrastructure or data engineering, along with strong proficiency in Python and ML frameworks. You'll be working in a collaborative environment, either in Bellevue, WA or Menlo Park, CA, with competitive compensation ranging from $157,000 to $185,000 USD.
Robinhood offers an impressive benefits package including comprehensive health coverage, mental health support, and family-friendly policies. The company maintains a vibrant office culture with catered meals and various lifestyle perks. This is an excellent opportunity for engineers passionate about building reliable platforms that drive company-wide impact while contributing to Robinhood's mission of making finance accessible to all.
The role combines technical challenges with meaningful impact, offering the chance to work on cutting-edge AI infrastructure while supporting Robinhood's goal of democratizing finance. You'll be part of a team that values innovation, collaboration, and technical excellence, while enjoying the benefits of working for a leading fintech company that's reshaping the financial industry.
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
- 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
- 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
Jobs Related To Robinhood Software Engineer, AI Infrastructure
