Snowflake

AI Research Scientist – Reinforcement Learning

Snowflake

full-time

Posted on:

Location: California • 🇺🇸 United States

Visit company website
AI Apply
Apply

Salary

💰 $195,000 - $250,000 per year

Job Level

Mid-LevelSenior

Tech Stack

PythonPyTorchTensorflow

About the role

  • Conduct advanced research in reinforcement learning, focusing on RLHF, DPO, PPO, and multi-agent systems
  • Develop custom solutions for real-world practical applications and challenging problem domains
  • Develop advanced reasoning models to enhance logical and contextual understanding for tasks like code generation and structured decision-making
  • Develop and fine-tune large language models, exploring post-training optimization techniques
  • Create and curate post-training data from synthetic data processing or human-annotated data pipelines
  • Collaborate with cross-functional teams to integrate research findings into real-world applications and products
  • Publish high-quality research papers in top-tier conferences and journals (e.g., NeurIPS, ICML, ICLR, ACL)
  • Stay up-to-date with latest developments in AI, RL, and LLMs and identify opportunities for innovation

Requirements

  • PhD in Computer Science, Machine Learning, Artificial Intelligence, or a closely related field (or equivalent research experience)
  • Deep understanding of reinforcement learning algorithms, including RLHF, DPO, PPO, and multi-agent systems
  • Strong expertise in large language models, including fine-tuning and post-training optimization techniques
  • Proven track record of publishing in top-tier AI conferences or journals (e.g., NeurIPS, ICML, ICLR, ACL)
  • Proficiency in programming languages and frameworks commonly used in AI research, such as Python, TensorFlow, or PyTorch
  • Excellent problem-solving skills, with a strong analytical and mathematical foundation
  • Ability to work both independently and collaboratively in a fast-paced research environment
  • Preferred: Experience with large-scale training and deployment of RL or LLM models
  • Preferred: Customer solution experience across vertical domains like healthcare and finance
  • Preferred: Familiarity with distributed computing and efficient training paradigms
  • Preferred: Experience developing reasoning models for mathematical problem-solving, code generation, or structured decision-making
  • Preferred: Background in human-computer interaction or related interdisciplinary fields
Salesforce

Research Scientist – Salesforce AI Research

Salesforce
Mid · Seniorfull-time$137k–$276k / yearCalifornia, New York · 🇺🇸 United States
Posted: 1 day agoSource: salesforce.wd12.myworkdayjobs.com
PythonPyTorchTensorflow
Allstate Northern Ireland

PhD Intern, AI Research Scientist

Allstate Northern Ireland
Entryinternship$119k–$162k / yearCalifornia · 🇺🇸 United States
Posted: 3 days agoSource: autodesk.wd1.myworkdayjobs.com
KerasPythonPyTorchScikit-LearnTensorflow
Autodesk

PhD Intern, AI Research Scientist

Autodesk
Entryinternship$119k–$162k / yearCalifornia · 🇺🇸 United States
Posted: 3 days agoSource: autodesk.wd1.myworkdayjobs.com
KerasPythonPyTorchScikit-LearnTensorflow
Topstep

Machine Learning Researcher

Topstep
Mid · Seniorfull-time$170k–$199k / year🇺🇸 United States
Posted: 4 days agoSource: boards.greenhouse.io
PythonSQL
Distyl AI

Applied AI Researcher, AI Systems

Distyl AI
Mid · Seniorfull-timeCalifornia · 🇺🇸 United States
Posted: 8 days agoSource: jobs.ashbyhq.com