Member of Technical Staff, Machine Learning Engineer

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

ABOUT US

Edison Scientific builds and deploys AI scientist agents to accelerate science and the development of new medicines. We are an ambitious team run by scientists and engineers from leading institutions across biology, physics, chemistry, and AI.

ABOUT THE ROLE

As a Member of Technical Staff, Machine Learning Engineer at Edison Scientific, you play a central role in building the models and agents that accelerate scientific discovery. You will work on both cutting edge research and practical engineering, bridging advanced machine learning concepts with robust, reliable software that real scientists depend on.

KEY RESPONSIBILITIES

• Interpret qualitative challenges in building AI agents for science as well-formulated optimizable problems
• Build appropriate environments in which to train and deploy AI agents that solve scientific tasks
• Work with scientists to formulate training data pipelines, and scale them, ensuring observability and reproducibility
• Lead training of large-scale LLM-based systems, including building internal infrastructure to improve the efficiency of experimentation and production training runs
• Build efficient and flexible inference infrastructure, supporting complex sampling algorithms and custom architectures
• Develop and extend our experimentation platform for internal tools and projects.
• Collaborate closely with a multidisciplinary team of AI researchers, chemists, biologists, fostering an environment of innovation and discovery.

REQUIRED QUALIFICATIONS

• 5-10+ years of strong track record of work in applied ML research and application of ML methods to solving real-world problems
• Experience working across the ML lifecycle: data pipelines and provenance, model training, model deployment, and validation in production systems.
• Fluency in PyTorch, Jax or equivalent framework.
• Demonstrated experience with experimentation in academic or industry settings.
• Strong programming expertise with the capability to adapt to various technical challenges in the data, ML, and LLM software stack.

PREFERRED QUALIFICATIONS

• PhD in Machine Learning, Computer Science, or other quantitative field
• Familiarity with leveraging and managing distributed computing resources
• Background architecting complex distributed systems

Salary

$240,000 - $350,000 Offers equity

WHY JOIN US?

We're a fast-moving, mission-driven culture where smart people do their best work and actually enjoy doing it. We also offer full-time employees the following:

• Competitive salary and equity
• Full healthcare coverage; we pay 100% of premiums for you and your dependents
• Support for growing families, including a yearly new parent stipend and fertility coverage through Carrot
• Mental health support through Rula, our in-network therapist and psychiatrist network with fast availability
• 12 weeks of paid parental leave for maternity, paternity, and adoption
• Pet care support with a yearly employer-funded stipend for your animal companions
• Commuter benefits so you can pay for transit and parking with pre-tax dollars
• 401(k) company matching
• $300 health and wellness benefit quarterly
• Lunch is on us every day you're in the office, and dinner is on us when you're working late
• Regular team off-sites and company events

Edison Scientific is an equal opportunity employer. We do not discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic under applicable law.