AI Research Scientist, New Grad – Agents & Reinforcement Learning

The available sources do not contain information about the company name for rounx.com.; United States; On-site; USD 140,000 - 190,000; Full time

Job summary

Snowflake is seeking a New Grad AI Research Scientist to join its AI Research team, focusing on agents and reinforcement learning, including designing agentic frameworks with self-improvement loops and building auto-research and coding agents.

The ideal candidate holds a PhD in CS or ML with expertise in reinforcement learning, LLM post-training, and agentic architectures; strong Python and deep learning skills (PyTorch or JAX) and solid mathematical abilities.

Qualifications

• PhD required in CS/ML or related field.
• Expertise in reinforcement learning, LLM post-training, and agentic architectures.
• Strong mathematical and analytical abilities.

Responsibilities

• developing multi-agent systems
• curating training data pipelines
• publishing findings at top AI conferences

Skills

Python

Reinforcement learning

LLM post-training

Agentic architectures

Mathematical/analytical ability

Education

PhD in Computer Science, ML, or related field

Tools

PyTorch

JAX

Job description

Snowflake is seeking a New Grad AI Research Scientist to join its AI Research team, focusing on agents and reinforcement learning. The role involves designing agentic frameworks with self-improvement loops, building auto-research and coding agents, and conducting research in RLHF, DPO, and PPO to align agentic behaviors.

Responsibilities

• developing multi-agent systems
• curating training data pipelines
• publishing findings at top AI conferences

Candidates should hold a PhD in Computer Science, Machine Learning, or a related field, with expertise in reinforcement learning, LLM post-training, and agentic architectures.

Proficiency in Python and deep learning frameworks like PyTorch or JAX is required, along with strong mathematical and analytical skills.

Prior publications in AI/ML are a plus.

The ideal candidate thrives in fast-paced environments, embraces experimentation, and aims to shape the future of enterprise AI through impactful research.