A

AI Researcher – Founding AI Team (Chemistry)

Visa Sponsorship
Apply Now

Leads large-scale model training and reinforcement learning systems for chemistry synthesis planning. Requires 5+ years of experience in distributed training, PyTorch/JAX, and frontier AI research. Must own end-to-end experiments from research to deployment.

Key Responsibilities
Lead large-scale model training and post-training experiments for chemistry synthesis planning
Develop pretraining, fine-tuning, synthetic data, and distillation pipelines
Build reinforcement learning systems that learn from experimental feedback
Design benchmarks, evaluations, and reproducible experiments
Run distributed training using frameworks like FSDP, DeepSpeed, or Megatron
Investigate model performance, identify failure modes, and improve reliability
Collaborate with chemists and product engineers to deploy AI research in practical workflows
Technical Skills Required
Distributed training frameworks (FSDP, DeepSpeed, Megatron) PyTorch/JAX Reinforcement learning
Benefits & Perks
$200,000–$300,000 base salary + meaningful equity
Visa sponsorship for H-1B, TN, O-1 (including transfers/renewals)
Competitive compensation
Founding team ownership of meaningful parts of training strategy
Collaboration with chemists, researchers, and product engineers
Real-world deployment of research models in industrial applications
Nice to Have
Experience with RL using verifiable rewards, sparse/delayed feedback
Familiarity with molecular or laboratory data
Experience with uncertainty modeling or active learning
Publications, open-source contributions, or benchmark/synthetic-data pipeline experience
Track record of deploying complex research into real-world models

Job Description

Location: San Francisco, CA

Working model: Full-time, in-office (5 days a week)

Working hours: 9 AM – 5 PM

Salary: $200,000–$300,000 base + meaningful equity

Experience: 5+ years (doctoral research experience counts)

Visa sponsorship: Available for US citizens, Green Card holders, H-1B, TN and O-1 (including transfers and renewals)

About the company

A Series B, AI-native chemical manufacturer based in San Francisco is using AI, advanced laboratories and a global manufacturing network to serve more than 100 enterprise customers.

The role

Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

Visas, Work Permits & Legal Insights

Stay informed about visa types, work permits, and immigration updates for international professionals.

Browse all articles

Subscribe to our newsletter

New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.