Senior Machine Learning Engineer, Gen AI

Robinhood • Menlo Park, CA

6h • $163,000 - $245,000 • Hybrid

About The Position

Robinhood is building an elite team applying frontier technologies to the world’s biggest financial problems. The AI Research and Development team's mission is to provide scalable data and model-driven decision-making solutions to various business functions at Robinhood. They aim to create a personalized user experience by helping users discover and engage with valuable products and features. To accelerate progress, they are also building an accessible model development platform to democratize machine learning practices throughout the company. This role is based in their Menlo Park, CA or Bellevue, WA office(s), with in-person attendance expected at least 3 days per week. Robinhood believes in the power of in-person work to accelerate progress, spark innovation, and strengthen community.

Requirements

• 5+ years of applied ML experience productionizing ML models
• 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

Nice To Haves

• Experience with Spark, Kafka, and Kubernetes is also desirable
• Ideally you have experience in the Finance sector

Responsibilities

• Evaluate cutting-edge technologies, including transformer-based model architectures and large foundational models, to identify solutions for Robinhood-specific problems.
• 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.
• Design reinforcement learning algorithms and apply multi-armed bandit strategies to optimize decision-making in dynamic environments.
• Design and conduct A/B tests to assess the performance of different machine learning models, including setting up test environments, monitoring performance, and analyzing results.
• Analyze experimental data to extract actionable insights using statistical techniques to validate findings.
• Collaborate with 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.
• Build reusable libraries for common machine learning practices.
• Offer support and guidance on the usage of these tools.
• Maintain comprehensive documentation of libraries, models, experiments, and findings.

Benefits

• 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

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed