Machine Learning Engineer, GenAI
RobinhoodJoin Our Team as a Machine Learning Engineer, GenAI
Robinhood is seeking a dedicated and ambitious Machine Learning Engineer to accelerate the development and expansion of products powered by Generative AI (GenAI) 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 GenAI team is devoted to bridging the transition of GenAI & ML modeling work into production-grade applications. Robinhood operates at the intersection of data-driven insights and technological innovation.
What You’ll Do
As a Machine Learning Engineer, you will focus on leveraging and optimizing Large Language Models (LLMs) along with the implementation of advanced AI technologies. 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 and accuracy.
- Evaluating Model Performance: Conduct rigorous evaluations of LLMs, assessing effectiveness, efficiency, and business alignment.
- Integration of Advanced AI Technologies: Implement Retrieval-Augmented Generation (RAG), function calling, and code interpreter technologies to enhance the capabilities of Large Language Models.
- Research and Development: Stay abreast of the latest advancements in machine learning, particularly in LLMs, LLM agents, and large-scale neural network training.
- 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 benefits
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.
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.
Benefits Extracted with AI
- Medical insurance
- Commuter benefits
- 100% paid health insurance for employees
- 90% coverage for dependents
- Annual lifestyle wallet
- Family forming and fertility benefits
- Dedicated mental health support
- Generous time away including company holidays, paid time off, sick time, parental leave
- Catered meals
- Fully stocked kitchens
Similar jobs
Last update: 23 minutes ago
Machine Learning Engineer with AI/ML Experience
Join us as a Machine Learning Engineer to develop AI/ML models and applications. Work remotely with top-tier companies.
Research Engineer, Language - Generative AI
Join Meta as a Research Engineer in Generative AI, focusing on large language models and NLP.
Founding Applied AI Engineer
Join LlamaIndex as a Founding Applied AI Engineer to build and deploy LLM applications. Competitive salary and equity offered.
Founding AI Engineer
Join LlamaIndex as a Founding AI Engineer to shape the future of LLM applications with cutting-edge AI projects.
Senior Machine Learning Engineer
Join Olo as a Senior Machine Learning Engineer to build and scale ML models for the restaurant industry. Remote work available.
Machine Learning Engineer
Join Ema Unlimited as a Machine Learning Engineer in SF Bay Area, working on cutting-edge AI solutions with a focus on NLP and ML technologies.
Machine Learning Engineer - Ads
Join as a Machine Learning Engineer focusing on Ads, developing predictive models in a hybrid role in New York.
Machine Learning Engineer
Join Snap Inc. as a Machine Learning Engineer in Los Angeles. Develop and deploy ML models to enhance user experience. Competitive salary and benefits.
Deep Learning Researcher
Join DataRobot as a Deep Learning Researcher to advance generative AI capabilities and integrate them into product offerings.
Senior Software Engineer, Machine Learning
Join Niantic as a Senior Software Engineer in Machine Learning to enhance products using generative AI technologies.
AI Research Scientist - Generative AI Red Teaming
Join Meta as an AI Research Scientist focusing on Generative AI Red Teaming, advancing AI responsibly.
AI Solutions Software Engineer
Join DwellFi as an AI Solutions Software Engineer to develop innovative AI solutions using LangChain or Llama.
Machine Learning Engineer - University Graduate 2025
Join Adobe as a Machine Learning Engineer to develop AI technologies for Photoshop and Digital Imaging.
Engineering Manager (Hands On)
Join Robin AI as an Engineering Manager in NYC, leading a team in a hybrid role with Python and React expertise.
Senior Applied AI Researcher
Join CHAI: AI Platform as a Senior Applied AI Researcher to optimize and innovate AI solutions in a high-growth environment.
Founding AI Engineer, Backend
Join LlamaIndex as a Founding AI Engineer, Backend to build scalable cloud services for LLM applications.
Internship Software Engineer - Machine Learning Infrastructure
Join Tesla as an Internship Software Engineer in Machine Learning Infrastructure. Work on AI infrastructure and neural network scaling.
AI Research Engineer, New PhD Graduate
Join Duolingo as an AI Research Engineer to solve complex problems and innovate in AI and data science.
Mid-Level AI Researcher
Join Leonardo.Ai as a Mid-Level AI Researcher to develop and refine AI models, focusing on model training and optimization.
Machine Learning Engineer
Join PayPal as a Machine Learning Engineer to develop advanced ML solutions for product and marketing strategies.
Founding Applied AI Engineer
Join Argon AI as a Founding Applied AI Engineer to lead AI initiatives in pharma, focusing on domain-specific AI and RAG systems.
Senior Software Engineer, Machine Learning
Join as a Senior Software Engineer in Machine Learning, working remotely to build ML-driven products for user engagement.
Senior ML Infrastructure Engineer
Join CHAI: AI Platform as a Senior ML Infrastructure Engineer to build and scale ML systems in Palo Alto.
Machine Learning Scientist, NLP
Join Abridge as a Machine Learning Scientist specializing in NLP to advance healthcare AI solutions.