The business is seeking to provide a robust, agile, and centralized AI platform to empower teams across Robinhood to rapidly build, deploy, and iterate on high-quality Machine Learning (ML) and Generative AI (GenAI) applications at scale.
Requirements
- 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.
Responsibilities
- 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.
Other
- 2+ years of software engineering experience, ideally within ML infrastructure, data engineering, or model operations.
- A collaborative mindset with an interest in building reliable platforms that support company-wide impact.
- 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.
