Job Description

Make more Creative Job Post Content< AI Research Scientist - Reinforcement Learning and Large Language Models (LLMs)

 - Responsibilities:

At Snowflake, we are on a mission to redefine how AI and ML shape our world. As the leader in secure and unified data platforms, we empower organizations to build generative AI applications with fully managed, enterprise-grade LLMs, all while ensuring unmatched governance and security.

We’re looking for an AI Research Scientist with deep expertise in Reinforcement Learning (RL) and Large Language Models (LLMs) to join our cutting-edge research team. This is your chance to drive groundbreaking innovation, from RLHF to advanced reasoning models, and help shape the future of AI on the Snowflake platform.

Your Mission

As an AI Research Scientist, you’ll work at the intersection of theoretical brilliance and practical application, helping Snowflake push the boundaries of AI innovation. Your work will directly influence Snowflake’s AI Data Cloud by advancing the capabilities of LLMs and RL-driven applications.

What You’ll Do

• Pioneering Research:
  Conduct advanced research in reinforcement learning (e.g., RLHF, DPO, PPO, multi-agent systems) to solve complex, real-world problems.

• Revolutionizing Reasoning Models:
  Develop models that enhance logical, contextual, and structured reasoning for tasks like code generation, mathematical problem-solving, and decision-making.

• Fine-Tuning LLMs:
  Explore post-training optimization techniques to enhance model performance, efficiency, and scalability.

• Data Innovation:
  Create and optimize data pipelines, from synthetic data processing to human-annotated datasets, to fuel advanced AI systems.

• Cross-Functional Impact:
  Collaborate with product, engineering, and other research teams to bring cutting-edge findings into real-world AI applications.

• Thought Leadership:
  Publish in top-tier conferences like NeurIPS, ICML, ACL, and ICLR, contributing to the global AI research community.

• Stay Ahead:
  Keep pace with the latest advancements in AI, RL, and LLMs, identifying emerging opportunities for innovation.

What You Bring

Core Qualifications

• Academic Excellence:
  PhD in Computer Science, Machine Learning, AI, or a related field (or equivalent research experience).

• Expertise in RL:
  In-depth knowledge of reinforcement learning techniques, including RLHF, DPO, PPO, and multi-agent systems.

• Mastery of LLMs:
  Strong experience with fine-tuning, post-training optimization, and deploying large-scale language models.

• Track Record of Innovation:
  Published high-quality research in top-tier AI conferences or journals.

• Technical Proficiency:
  Advanced skills in Python, TensorFlow, and PyTorch, with experience in distributed computing and efficient training paradigms.

• Analytical & Collaborative Skills:
  Strong problem-solving abilities and the capacity to thrive in collaborative, fast-paced environments.

Preferred Skills

• Experience with large-scale RL or LLM model deployment.
• Expertise in developing reasoning models for structured tasks (e.g., code generation or decision-making).
• Familiarity with human-computer interaction and interdisciplinary fields.

Why Snowflake?

Impactful Work

Contribute to a platform transforming how organizations harness AI and ML to create real-world impact.

Cutting-Edge Innovation

Work on advanced AI research at the forefront of RLHF, LLMs, and reasoning models.

Culture of Excellence

Join a team of brilliant minds passionate about challenging conventional thinking and driving the pace of innovation.

Competitive Benefits

• Competitive salary with bonus, equity, and benefits.
• Flexible work arrangements to support your growth and balance.
• Opportunities to publish your research and grow your reputation in the global AI community.