Senior Machine Learning Engineer, AI Infra
Core
Building a robust, centralized AI platform to empower teams across Robinhood by defining how AI gets built and run, focusing on reliability, scalability, and craft.
Role type
Senior IC machine learning infrastructure engineer (platform)
Builds
Scalable systems for deploying, monitoring, and managing ML models in production, including model serving, feature stores, and training infrastructure.
Domain
Financial technology + ML infrastructure
Deliverable
production ML models
Required skills
ML infrastructure, distributed systems, production ML workflows, Python, C++, TensorFlow, PyTorch, Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton, vector databases, embedding models, large-scale search systems
Preferred skills
Advanced degree in CS/SE, experience with large-scale search systems using dense vector search
Technologies
Ray, Kubeflow, SageMaker, TensorFlow Serving, Triton, Qdrant, ChromaDB, Elasticsearch
Responsibilities
Lead architecture and end-to-end delivery of scalable systems for ML model deployment and monitoring; Own technical direction for model serving, feature store, and ML observability; Drive cross-functional partnerships to streamline ML workflows; Evolve and scale the feature store for real-time and batch use cases; Define and implement robust observability standards; Manage and optimize cloud compute resources on AWS; Contribute to technical strategy and mentor engineers
Seniority
Senior, hands-on IC
