Snorkel AI
Research Scientist - RL Training
Redwood City, CA (Hybrid); San Francisco, CA (Hybrid); United States (Remote)
Sponsorship not specifiedDetected 15 days ago
PythonNode.jsAWSGCPCloud PlatformsKubernetesMachine LearningPyTorchData EngineeringLLMsResearchLeadership
About the role
- We're looking for a Research Scientist to work on reinforcement learning for training and aligning large language models.
- Prototype and iterate on end-to-end RL training recipes that inform what data Snorkel ships as part of its data-as-a-service deliveries.
- Work closely with research scientists, ML engineers, and delivery teams to translate RL research into customer-ready data products.
Responsibilities
- Design and build data pipelines that generate high-quality training signal for RL workflows, including AI-assisted data annotation and curation data pipelines to improve model generalization to unseen benchmarks.
- the data you use to build AI is the key to achieving differentiation, high performance, and production-ready systems.
- We work with some of the world's largest organizations to empower scientists, engineers, financial experts, product creators, journalists, and more to build custom AI with their data faster than ever before.
Nice to have
- Experience training or fine-tuning 30B+ large language models at scale, including familiarity with distributed training infrastructure.
- Strong proficiency in Python and ML frameworks, especially PyTorch and HuggingFace and hands-on experience with RL frameworks such as Verl and SkyRL.
- Familiarity with ML infrastructure and cloud platforms and tools (AWS, GCP, Kubernetes, Slurm, etc.)
- experience with large-scale RL training pipelines a strong plus.
- Comfort operating in a high-iteration environment with open-ended research questions and shifting, customer-driven technical constraints.
- exceptional industry experience considered.
- Be Your Best at Snorkel
- As a member of our team, you'll have meaningful opportunities to shape priorities and initiatives, influence key strategic decisions, and directly impact our ongoing success.
Apply directly at Snorkel AI →Create a free account for alerts like thisView Snorkel AI immigration profile
This listing is sourced directly from Snorkel AI's careers page and normalized into a canonical job model.