Principal Machine Learning Engineer

Posted 25 Days Ago
Easy Apply
San Francisco, CA, USA
In-Office
185K-275K Annually
Mid level
Artificial Intelligence • Software • Database
The Role
As a Machine Learning Engineer, you will build models and agents to accelerate scientific discovery, collaborating with researchers and developing efficient infrastructure.
Summary Generated by Built In
About

Edison Scientific builds and commercializes AI agents for science. Scientific discovery moves too slowly, and autonomous AI agents are how we intend to fix that. We're assembling a team of top researchers and engineers across AI and biology to build an AI scientist.

Role

As a Principal 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.

This role is on-site at our San Francisco office in the Dogpatch neighborhood. Our office is a converted warehouse with high ceilings, open space, and a team excited about what we’re building.

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.
Qualifications
  • 8-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.
Bonus points for
  • PhD in Machine Learning, Computer Science, or other quantitative field
  • Familiarity with leveraging and managing distributed computing resources
  • Background architecting complex distributed systems

Salary

$275,000 - $350,000 •  Offers equity

Why join us?

  • 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
  • 401(k) company matching
  • $300 health and wellness benefit
  • Lunch is on us every day you're in the office, and dinner is on us when you're working late
  • Regular team offsites and company events
  • A fast-moving, mission-driven culture where smart people do their best work and actually enjoy doing it

Top Skills

Jax
PyTorch
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, California
47 Employees
Year Founded: 2025

What We Do

Spun out from FutureHouse in 2025, Edison Scientific accelerates discovery and innovation across the sciences. Our platform empowers researchers to move from question to breakthrough faster than ever, automating literature synthesis, data analysis, and molecular design. At its core is Kosmos, our AI scientist, capable of running hundreds of research tasks in parallel. It transforms raw datasets into comprehensive, validated reports—compressing months of work into a single run. With Edison Scientific, scientists remain in control, using AI to amplify their expertise and accelerate discovery at unprecedented speed.

Similar Jobs

General Motors Logo General Motors

Machine Learning Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Hybrid
Mountain View, CA, USA
165000 Employees

Snap Inc. Logo Snap Inc.

Machine Learning Engineer

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
3 Locations
5000 Employees
235K-414K Annually

ServiceNow Logo ServiceNow

Machine Learning Engineer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Santa Clara, CA, USA
28000 Employees
236K-413K Annually

d-Matrix Logo d-Matrix

Test Engineer

Artificial Intelligence • Machine Learning • Software
Hybrid
Santa Clara, CA, USA
102 Employees
180K-300K Annually

Similar Companies Hiring

Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY
Bellagent Thumbnail
Artificial Intelligence • Machine Learning • Business Intelligence • Generative AI
Chicago, IL
20 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account