Senior Machine Learning Engineer, Agentic

Robinhood
Menlo Park, CA, United States
2 months ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Experience level
Expert
Experience required
2 years minimum
Compensation
$209,000.0 - $245,000.0
Working hours
Regular working hours
Job source

Tech stack

A/B Testing Artificial Intelligence Data Analysis Experimental Data Python (Programming Language) Machine Learning Recommender Systems Tensorflow SQL Databases Reinforcement Learning Pytorch Apache Spark
+3 more
Kubernetes Xgboost Apache Kafka

Job description

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.

Requirements

Do you have experience in Machine learning frameworks?, * 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.

Benefits & conditions

Pulled from the full job description

  • Parental leave
  • Health insurance
  • 401(k) matching
  • Paid time off
  • Disability insurance
  • Paid holidays, * 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.

About the company

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.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on indeed.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:01 min

Exploring foundational expertise in traditional optimization and machine learning

Eric Enge · Coffee With Developers

2:35 min

Preventing remote code execution in PyTorch models

Balázs Kiss · WWC 2023

1:09 min

Configuring synthetic data for safe interactive programming

Mingshen Sun Mingshen Sun · WWC 2024

1:39 min

Fundamentals of tensors and the TensorFlow library

Håkan Silfvernagel · LIVE

3:53 min

Architecting machine learning projects with the PAI platform

Qiyang Duan · LIVE

7:10 min

Exploring pathways into the machine learning engineering field

Jose Luis Latorre Millas · LIVE

Videos

See all

Related articles

See all