Senior Machine Learning Engineer, Agentic

About the team + role

We are building an elite team, applying frontier technologies to the world’s biggest financial problems. We’re looking for bold thinkers. Sharp problem-solvers. Builders who are wired to make an impact. Robinhood isn’t a place for complacency, it’s where ambitious people do the best work of their careers. We’re a high-performing, fast-moving team with ethics at the center of everything we do. Expectations are high, and so are the rewards.

The Agentic team at Robinhood builds and ships production AI agents that power the next generation of AI financial products. Our mission is to rapidly build, evaluate, and deploy high-performance AI agents on production-grade infrastructure, strong evaluation and observability baked in, and continuous optimization support. 

This role is based in our Menlo Park, CA or Bellevue, WA office(s), with in-person attendance expected at least 3 days per week. 

At Robinhood, we believe in the power of in-person work to accelerate progress, spark innovation, and strengthen community. Our office experience is intentional, energizing, and designed to fully support high-performing teams.

What you'll do

As a Machine Learning Engineer on our team, your primary focus will be on the implementation and evaluation of machine learning algorithms through rigorous experimentation and testing methodologies. Your responsibilities will include:

• AI and ML Research: Evaluate cutting technologies, including but not limited to, transformer based model architecture and large foundational models to identify solutions for Robinhood specific problems.
• Model Development and Implementation: Develop and implement scalable machine learning models focusing on advanced ranking and recommendation systems, including expertise in Collaborative Filtering, Content-Based Filtering, and Hybrid models, alongside proficiency in Learning to Rank (LTR) techniques for effective prioritization. Additionally, design reinforcement learning algorithms and apply multi-armed bandit strategies to optimize decision-making in dynamic environments, balancing exploration and exploitation.
• A/B Testing and Experimentation: Design and conduct A/B tests to assess the performance of different machine learning models. This includes setting up the test environment, monitoring performance, and analyzing results.
• Data Analysis and Insight Generation: Analyze experimental data to extract actionable insights. Use statistical techniques to validate the findings and ensure their relevance and accuracy.
• Cross-Functional Collaboration: Work closely with other engineering teams, data scientists, and the marketing team to integrate machine learning models into the product and ensure they meet business requirements. Present results to different stakeholders.
• Tooling and Documentation: Build reusable libraries for common machine learning practices. Offer support and guidance to the usage of these tools. Maintain comprehensive documentation of libraries, models, experiments, and findings.

What you bring

• 5+ years of applied ML experience productionizing ML models with 2+ years focused on recommendations, ranking or personalization projects.
• A fervent interest in exploring and applying AI and ML technologies.
• Strive to solve sophisticated engineering problems that drive business objectives.
• Solid technical foundation enabling active contribution to the design and execution of projects and ideas.
• Familiarity with architectural frameworks of large, distributed, and high-scale ML applications.
• Hands-on experience in classical ML techniques with tabular data as well as modern techniques with sequential data
• Proven experience in ML with a focus on ranking, recommendation systems, multi-objective optimization, and reinforcement learning.
• Proficiency in Python, SQL, XGboost, PyTorch/TensorFlow.
• Experience with Spark, Kafka, and Kubernetes is also desirable.
• Ideally you have experience in the Finance sector.

What we offer

• Challenging, high-impact work to grow your career
• Performance driven compensation with multipliers for outsized impact, bonus programs, equity ownership, and 401(k) matching
• Best in class benefits to fuel your work, including 100% paid health insurance for employees with 90% coverage for dependents
• Lifestyle wallet - a highly flexible benefits spending account for wellness, learning, and more
• Employer-paid life & disability insurance, fertility benefits, and mental health benefits
• Time off to recharge including company holidays, paid time off, sick time, parental leave, and more!
• Exceptional office experience with catered meals, events, and comfortable workspaces.