Senior AI Research Scientist: Scientific ML & LLMs

Edison; San Francisco (CA); On-site; USD 210,000 - 350,000; Full time

Benefits offered by this job

Full healthcare coverage

Parental leave stipend

Mental health support (Rula)

401(k) matching

Office lunch provided

Job summary

Edison Scientific in San Francisco, CA, seeks an experienced Member of Technical Staff, AI Research Scientist to advance models for accelerating scientific discovery. This on-site role blends cutting‑edge research with practical deployment, collaborating with biologists, chemists, and engineers to push AI for science.

You will own end‑to‑end projects, develop scalable experimentation platforms, and contribute to production systems.

Qualifications

• 5-10+ years of strong track record of ML research on real-world problems in reinforcement learning, representation learning, and Bayesian methods.
• Fluency in PyTorch, Jax, or equivalent framework.
• Experience with large-scale training and experimentation technologies (e.g. Ray, Verl, Megatron-LM).
• Experience working with data and ML systems: EDA, feature engineering, model development, deployment, and production validation.
• Familiarity with modeling in PyTorch, Jax or equivalent framework.
• Demonstrated experience with experimentation in academic or industry settings.
• Strong programming expertise across data, ML, and LLM software stacks.

Responsibilities

• Interpret challenges in building AI agents for science as well‑formulated statistical problems.
• Conduct research in post-training of LLMs on challenging scientific reasoning tasks to exceed expert human capacity.
• Investigate sampling techniques focusing on exploration and quality‑diversity.
• Own research projects end‑to‑end from scoping with scientists to deploying outcomes with engineers.
• Develop and extend our experimentation platform for internal tools and projects.
• Collaborate with a multidisciplinary team of AI researchers, engineers, biologists, and chemists.

Skills

ML Research

Reinforcement Learning

Representation Learning

Bayesian Methods

PyTorch

Jax

Large-scale Training

Experimentation

Software Engineering

Education

PhD in Machine Learning/Computer Science

Tools

PyTorch

Jax

Ray

Megatron-LM

Job description

Edison Scientific in San Francisco, CA, seeks an experienced Member of Technical Staff, AI Research Scientist to advance models for accelerating scientific discovery. This on-site role blends cutting‑edge research with practical deployment, collaborating with biologists, chemists, and engineers to push AI for science.

You will own end‑to‑end projects, develop scalable experimentation platforms, and contribute to production systems.