Senior ML Engineer — Scientific AI & LLM Systems

edisonscientific.com; San Francisco (CA); On-site; USD 240,000 - 350,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Fertility coverage

Mental health support

Equity

401(k) matching

Transit benefits

Lunch provided

Dinner provided

Team off-sites

Job summary

Edison Scientific is seeking a Member of Technical Staff, Machine Learning Engineer to drive the development of models and agents accelerating scientific discovery. You'll bridge cutting-edge ML research with robust software that scientists rely on and collaborate with chemists, biologists, and AI researchers.

This is a hands-on role in a fast-moving, mission-driven team. You will lead training of large-scale systems, build scalable inference, and contribute to internal tooling and data

Qualifications

• 5-10+ years of applied ML research and real-world problem solving.
• Experience across ML lifecycle: data pipelines, training, deployment, and validation in production.
• Fluency in PyTorch, JAX or equivalent frameworks.
• Strong programming能力 and adaptability to ML, data, and LLM software stacks.

Responsibilities

• Interpret qualitative ML challenges and formulate them as optimizable problems.
• Build environments to train and deploy AI agents for scientific tasks.
• Collaborate with scientists to design data pipelines with observability and reproducibility.
• Lead training of large-scale LLM-based systems and improve experimentation infrastructure.
• Develop scalable inference infrastructure supporting complex sampling and architectures.
• Extend our experimentation platform for internal tools and projects.
• Collaborate with a multidisciplinary AI team to drive innovation.

Skills

Applied ML research

ML lifecycle

PyTorch

Python

Model deployment

Experimentation

Education

PhD in ML or CS

Tools

TensorFlow

Kubernetes

Docker

Job description

Edison Scientific is seeking a Member of Technical Staff, Machine Learning Engineer to drive the development of models and agents accelerating scientific discovery. You'll bridge cutting-edge ML research with robust software that scientists rely on and collaborate with chemists, biologists, and AI researchers.

This is a hands-on role in a fast-moving, mission-driven team. You will lead training of large-scale systems, build scalable inference, and contribute to internal tooling and data