R
AI

Machine Learning Engineer GenAI

Robinhood · Menlo Park, CA; New York, NY · $105k - $110k

Actively hiring Posted over 1 year ago

Role overview

Robinhood Markets was founded on a simple idea: that our financial markets should be accessible to all. With customers at the heart of our decisions, Robinhood and its subsidiaries and affiliates are 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.

With growth as the top priority...

The business is 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.

What you'll work on

  1. Development and Optimization of LLMs: Implement and fine-tune state-of-the-art Large Language Models for various applications, focusing on performance and accuracy.
  2. Evaluating Model Performance: Conduct rigorous evaluations of LLMs, assessing effectiveness, efficiency, and business alignment.
  3. Integration of Advanced AI Technologies: Implement Retrieval-Augmented Generation (RAG), function calling, and code interpreter technologies to enhance the capabilities of Large Language Models.
  4. Research and Development: Stay abreast of the latest advancements in machine learning, particularly in LLMs, LLM agents, and large-scale neural network training.
  5. Data and Model Parallel Training: Utilize data and model parallel training techniques for efficient handling of large-scale models.
  6. GPU Cluster Management for Training: Oversee extensive training jobs on GPU clusters, ensuring optimal resource utilization for complex tasks.
  7. Cross-Functional Collaboration and Leadership: Work with ML engineers, data scientists, and product teams, providing guidance and mentorship.
  8. Documentation and Reporting: Maintain detailed documentation of methodologies, models, and results, and communicate findings across the organization.

What we're looking for

  1. Advanced Degree: Master's or PhD in Computer Science, AI, Linguistics, or related fields, with a focus on machine learning and natural language processing.
  2. Experience with LLMs and PyTorch: Extensive experience with large language models and proficiency in PyTorch.
  3. Expertise in Parallel Training and GPU Cluster Management: Strong background in parallel training methods and managing large-scale training jobs on GPU clusters.
  4. Analytical and Problem-Solving Skills: Ability to address complex challenges in model training and optimization.
  5. Leadership and Mentorship Capabilities: Proven leadership in guiding projects and mentoring team members.
  6. Communication and Collaboration Skills: Effective communication skills for conveying technical concepts and collaborating with cross-functional teams.
  7. Innovation and Continuous Learning: Passion for staying updated with the latest trends in AI and machine learning.

Tags & focus areas

Used for matching and alerts on DevFound
Engineer Machine Learning Ai Pytorch