Staff Machine Learning Engineer, Generative AI
Menlo Park, CA
Full Time Senior-level / Expert USD 199K - 255K
- Remote-first
- @RobinhoodApp
- GitHub
- Website
Robinhood
Trade 30+ crypto at the lowest cost on average in the EU. Sign up today and get a reward of up to 1 BTC.Join a leading fintech company that’s democratizing finance for all.
Robinhood was founded on a simple idea: that our financial markets should be accessible to all. With customers at the heart of our decisions, Robinhood is lowering barriers and providing greater access to financial information. Together, we are building products and services that help create a financial system everyone can participate in.
As we continue to build...
We’re seeking curious, growth minded thinkers to help shape our vision, structures and systems; playing a key-role as we launch into our ambitious future. If you’re invigorated by our mission, values, and drive to change the world — we’d love to have you apply.
About the team + role
We are seeking a dedicated and ambitious individual to accelerate the development and expansion of products powered by Gen AI to democratize finance at an unprecedented pace. In this role, you'll play a key part in Robinhood’s forward trajectory, collaborating closely with our adept Data Science and Engineering teams. The Gen AI team is devoted to bridging the transition of Gen AI & ML modeling work into production-grade applications. Robinhood operates at the intersection of data-driven insights and technological innovation.
The role is located in the office location(s) listed on this job description which will align with our in-office working environment. Please connect with your recruiter for more information regarding our in-office philosophy and expectations.
What you'll do
In your role as a Staff Machine Learning Engineer, you will focus on the highest-impact, org-wide applications of LLMs to accelerate the company’s mission. Your key responsibilities will include:
- Development and Optimization of LLMs: Implement and fine-tune state-of-the-art Large Language Models for various applications, focusing on performance, accuracy, reliability and safety.
- Evaluating Model Performance: Conduct rigorous evaluations of LLMs, assessing effectiveness, efficiency, and business alignment.
- Integration of State-of-the-art GenAI “agents”: Retrieval-Augmented Generation (RAG), function calling, and code interpreter technologies to enhance the capabilities of Large Language Models. You are also able to design and implement novel extensions using the paradigms of Reinforcement Learning and self-play.
- Research and Development: Stay abreast of the latest advancements in machine learning, particularly in LLMs, LLM agents, and large-scale neural network training for the purposes of finetuning and model distillation.
- Data and Model Parallel Training: Utilize data and model parallel training techniques for efficient handling of large-scale models.
- GPU Cluster Management for Training: Oversee extensive training jobs on GPU clusters, ensuring optimal resource utilization for complex tasks.
- Cross-Functional Collaboration and Leadership: Work with ML engineers, data scientists, and product teams, providing guidance and mentorship.
- Documentation and Reporting: Maintain detailed documentation of methodologies, models, and results, and communicate findings across the organization.
What you bring
- Advanced Degree: Master's or PhD in Computer Science, AI, Linguistics, or related fields, with a focus on machine learning and natural language processing.
- Experience with LLMs and PyTorch: Extensive experience with large language models and proficiency in PyTorch.
- Expertise in Parallel Training and GPU Cluster Management: Strong background in parallel training methods and managing large-scale training jobs on GPU clusters.
- Analytical and Problem-Solving Skills: Ability to address complex challenges in model training and optimization.
- Leadership and Mentorship Capabilities: Proven leadership in guiding projects and mentoring team members.
- Communication and Collaboration Skills: Effective communication skills for conveying technical concepts and collaborating with cross-functional teams.
- Innovation and Continuous Learning: Passion for staying updated with the latest trends in AI and machine learning.
What we offer
- Market competitive and pay equity-focused compensation structure
- 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 benefit
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. The expected salary range for this role is based on the location where the work will be performed and is aligned to one of 3 compensation zones. This role is also eligible to participate in a Robinhood bonus plan and Robinhood’s equity plan. For other locations not listed, compensation can be discussed with your recruiter during the interview process.
Zone 1 (Menlo Park, CA; New York, NY; Bellevue, WA; Washington, DC)$217,000—$255,000 USDZone 2 (Denver, CO; Westlake, TX; Chicago, IL)$190,000—$224,000 USDZone 3 (Lake Mary, FL)$169,000—$199,000 USDClick here to learn more about Robinhood’s Benefits.
We’re looking for more growth-minded and collaborative people to be a part of our journey in democratizing finance for all. If you’re ready to give 100% in helping us achieve our mission—we’d love to have you apply even if you feel unsure about whether you meet every single requirement in this posting. At Robinhood, we're looking for people invigorated by our mission, values, and drive to change the world, not just those who simply check off all the boxes.
Robinhood embraces a diversity of backgrounds and experiences and provides equal opportunity for all applicants and employees. We are dedicated to building a company that represents a variety of backgrounds, perspectives, and skills. We believe that the more inclusive we are, the better our work (and work environment) will be for everyone. Additionally, Robinhood provides reasonable accommodations for candidates on request and respects applicants' privacy rights. To review Robinhood's Privacy Policy please review the specific policy applicable to the country where you are applying.
Tags: Computer Science Engineering Finance FinTech Generative AI GPU Linguistics LLMs Machine Learning Model training NLP PhD Privacy PyTorch Reinforcement Learning Research
Perks/benefits: Career development Competitive pay Equity Fertility benefits Flex vacation Health care Lunch / meals Parental leave Salary bonus Wellness
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Business Intelligence Engineer jobs
- Open Lead Data Analyst jobs
- Open Power BI Developer jobs
- Open Data Engineer II jobs
- Open Senior Business Intelligence Analyst jobs
- Open Marketing Data Analyst jobs
- Open Data Science Manager jobs
- Open MLOps Engineer jobs
- Open Junior Data Scientist jobs
- Open Data Scientist II jobs
- Open Business Intelligence Developer jobs
- Open Business Data Analyst jobs
- Open Product Data Analyst jobs
- Open Data Analytics Engineer jobs
- Open Sr Data Engineer jobs
- Open Data Analyst Intern jobs
- Open Principal Data Scientist jobs
- Open Sr. Data Scientist jobs
- Open Senior Data Architect jobs
- Open Data Engineering Manager jobs
- Open Junior Data Engineer jobs
- Open Big Data Engineer jobs
- Open Azure Data Engineer jobs
- Open Manager, Data Engineering jobs
- Open Research Scientist jobs
- Open GCP-related jobs
- Open Java-related jobs
- Open Data quality-related jobs
- Open ML models-related jobs
- Open Business Intelligence-related jobs
- Open Data management-related jobs
- Open Privacy-related jobs
- Open PhD-related jobs
- Open Data visualization-related jobs
- Open Deep Learning-related jobs
- Open Finance-related jobs
- Open NLP-related jobs
- Open PyTorch-related jobs
- Open TensorFlow-related jobs
- Open LLMs-related jobs
- Open APIs-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open Snowflake-related jobs
- Open Consulting-related jobs
- Open Hadoop-related jobs
- Open Kubernetes-related jobs
- Open Data governance-related jobs
- Open Databricks-related jobs
- Open Airflow-related jobs