Jobs Companies Chai Discovery Software Engineer, Platform & Inference

About this Software Engineer, Platform & Inference role at Chai Discovery

Chai Discovery · Onsite · San Francisco office

About Chai Discovery

Chai builds the design suite for molecules. We train frontier models that learn the underlying foundations of biochemical structure and interaction, so scientists can move faster and pursue targets that other methods cannot reach.

AI is reinventing life sciences the same way it reinvented software engineering, and Chai is at the forefront of this shift. Leading pharmaceutical companies like Eli Lilly, Pfizer, and Novartis are adopting our platform to power their drug discovery programs.

We value diverse perspectives and are ready to find greatness in unexpected places.

About the role

Platform engineers make Chai's models fast, cheap, and reliable at scale, and enable the outer loop that accelerates research: the infrastructure and software abstractions used to train, eval, and understand models.

You'll own the serving stack that turns our frontier models into a product scientists depend on: latency, throughput, GPU efficiency, batching, and autoscaling across a large multi-cloud GPU fleet. You'll also contribute to the work that enables turning raw models into product-ready pipelines, and the experiment and observability tooling that lets a researcher ship faster.

You've built high-performance services that developers love, moved ML systems into production at scale, and can see around corners before they become outages.

You'll work closely with the researchers who train the models, the product engineers who build on them, and the commercial team deploying them to the world's largest pharma companies.

About you

We index on systems judgment, ownership, and the scars that come from having run production infrastructure before. We're looking for engineers who get obsessed with hard problems and don't give up easily. We look for:

  • 4+ years building production systems, with real depth in performance, distributed systems, or ML serving

  • Experience optimizing model inference: GPU utilization, batching, quantization, caching, or kernel-level work

  • A platform mindset: you like building the tools and abstractions that make other engineers and researchers faster

  • End-to-end ownership of 24/7 systems, including observability, alerting, and incident response

  • Experience across both 0-to-1 buildouts and 1-to-n scale-ups, with an always-evolving playbook you bring wherever you go

  • The instinct to treat cost and efficiency as first-class constraints, not afterthoughts

A background in biology is not required. What makes the difference is technical excellence, curiosity about the domain, and grit.

We offer

The opportunity to work at the vanguard of AI research and frontier biology, with world-class people, on a mission that matters. We protect & promote a culture of high velocity and ownership. We compensate our team accordingly.

Ready to apply to Chai Discovery?
Apply to Chai Discovery

Similar jobs

Lyft
ML Software Engineer, ETA
Lyft
⚡ Apply early San Francisco, CA Hybrid $140,800–$176,000
● New 👁 Seen ✓ Applied 12h ago
Lyft
Senior ML Software Engineer, Mapping
Lyft
⚡ Apply early San Francisco, CA Hybrid $162,800–$203,500
● New 👁 Seen ✓ Applied 12h ago
Lyft
Staff Software Engineer
Lyft
⚡ Apply early San Francisco, CA Hybrid $176,000–$220,000
● New 👁 Seen ✓ Applied 12h ago
Mercury
Senior Software Engineer - Investments
Mercury
⚡ Apply early San Francisco, CA, New York, N... · location restricted $200,700–$250,900
● New 👁 Seen ✓ Applied 13h ago
Sydecar
Senior Software Engineer
Sydecar
⚡ Apply early San Francisco Office - Hybrid Hybrid $190,000–$255,000
● New 👁 Seen ✓ Applied 1d ago
Sydecar
Staff Software Engineer
Sydecar
⚡ Apply early San Francisco Office - Hybrid Hybrid $225,000–$310,000
● New 👁 Seen ✓ Applied 1d ago
Hayden AI
Senior Software Engineer - Device
Hayden AI
⚡ Apply early San Francisco HQ Office Hybrid $160,000–$260,000
● New 👁 Seen ✓ Applied 1d ago
Chime Financial, Inc
Software Engineer, Growth
Chime Financial, Inc
⚡ Apply early San Francisco, CA, USA Onsite
● New 👁 Seen ✓ Applied 1d ago
DriveWealth
Principal Software Development Engineer, Execution Services
DriveWealth
⚡ Apply early AMER-US-Remote; Austin, Texas,... Hybrid $250,000–$270,000
● New 👁 Seen ✓ Applied 1d ago

Sign up for suggestions tailored to the jobs you open and the searches you save.

More jobs at Chai Discovery

See all jobs at Chai Discovery →

Apply now
🤖

Whoa — hold up

JobsRadar was built for real people having a rough time in their job search — not for automated requests. You're clicking way too fast and you're now temporarily blocked.

Come back later. If you're genuinely job hunting, we've got your back — just act like a human.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Get an edge on your job hunt.

Join our Telegram channel for the stuff that helps you land the role — salary benchmarks, the weekly market pulse, and new-feature drops. No spam, just signal.

Join the channel — it's free