Snowflake · 4 months ago
AI Research Scientist - Reinforcement Learning
US-CA-Menlo Park
Full-time
Onsite
Senior Level
$195K/yr - $250K/yrSnowflake is a leading company in the field of AI and ML, seeking a talented AI Research Scientist specializing in Reinforcement Learning and Large Language Models. The role involves conducting advanced research, developing custom solutions, and collaborating with teams to integrate findings into real-world applications.
AnalyticsArtificial Intelligence (AI)Cloud Data ServicesData ManagementEnterprise SoftwareSoftware
Responsibilities
Conduct advanced research in reinforcement learning, focusing on areas such as RLHF (Reinforcement Learning with Human Feedback), DPO (direct preference optimization), proximal policy optimization (PPO), and multi-agent systems
Develop custom solutions for a variety of real-world practical applications and challenging problem domains
Develop advanced reasoning models to enhance logical and contextual understanding, with applications in tasks such as code generation and structured decision-making
Develop and fine-tune large language models, exploring post-training optimization techniques to enhance performance and efficiency
Familiar with post-training data creation from either 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 the latest developments in AI, RL, and LLMs, identifying opportunities for innovation and improvement
Qualification
Required
PhD in Computer Science, Machine Learning, Artificial Intelligence, or a closely related field (or equivalent research experience)
Deep understanding of reinforcement learning algorithms, including recent advances like 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
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
Have customer solution experience for different vertical domains, like health and finance
Possess customer solution experience across diverse vertical domains, including healthcare and finance
Familiarity with distributed computing and efficient training paradigms
Experience developing reasoning models, such as those for mathematical problem-solving, code generation, or structured decision-making tasks
Background in human-computer interaction or related interdisciplinary fields
Company
Snowflake
Snowflake is a cloud data platform that provides a data warehouse as a service designed for the cloud.
Founded in 2012
5001-10000 employees
H1B Sponsorship
Snowflake has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (428)
2024 (281)
2023 (154)
2022 (182)
2021 (113)
2020 (98)
Funding
Current Stage
Public CompanyTotal Funding
$2.03BKey Investors
Sequoia CapitalICONIQ GrowthAltimeter Capital
2022-04-19Post Ipo Equity· $621.46M
2020-09-16IPO
2020-02-07Series G· $479M
Leadership Team
Sridhar Ramaswamy
Chief Executive Officer
Benoit Dageville
Snowflake Co-founder, President of Product
Recent News
The Motley Fool
2026-01-09
2026-01-09
Tech in Asia
2026-01-09
Company data provided by crunchbase