Software Engineer, AI Infrastructure
Description For Software Engineer, AI Infrastructure
Robinhood Markets is seeking a Software Engineer for their AI Infrastructure team to help democratize finance for all. This role is crucial in building and maintaining the company's machine learning platform that enables teams across Robinhood to develop and deploy AI applications at scale. The position offers an opportunity to work on cutting-edge AI infrastructure, supporting both traditional ML and Generative AI applications.
The role involves designing and implementing scalable systems for ML model deployment, monitoring, and management in production environments. You'll collaborate closely with Data Scientists and Applied ML Engineers to create foundational systems that accelerate experimentation and improve model observability. Key responsibilities include developing feature stores, implementing robust monitoring systems, and optimizing cloud resources for ML workloads.
This is an ideal opportunity for an experienced software engineer with a strong background in ML infrastructure or data engineering. The position requires expertise in Python, familiarity with ML frameworks like TensorFlow or PyTorch, and experience with modern ML infrastructure tools. You'll be working in a collaborative environment where your contributions will directly impact Robinhood's mission of making financial markets accessible to everyone.
The compensation is highly competitive, with a base salary range of $157,000 - $185,000 USD for primary locations, complemented by comprehensive benefits including full health coverage, mental health support, and various lifestyle perks. The role offers the excitement of working at the intersection of fintech and AI, building systems that power the future of financial services.
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
