← Back to results

Software Engineer, Compute Infrastructure

Join Chai Discovery as a Software Engineer to optimize AI models for drug discovery in a fast-paced, innovative environment.

Location
San Francisco office
Compensation
Not disclosed
Level
senior
Type
full time

Posted by employer 11 months ago

First seen on Joblaze 1 month ago

Last verified on the company career page 6 hours ago

Apply at Chai Discovery → Save job Scanned from chaidiscovery.com

AI in the day-to-day

Chai builds models that learn biochemical structures to accelerate drug discovery.

Requirements

Experience
4+ years

Not disclosed in this posting: compensation, work arrangement, visa sponsorship.

Joblaze summary

In this role, the software engineer focuses on optimizing and maintaining the infrastructure that supports Chai Discovery's advanced AI models, ensuring they operate efficiently and reliably at scale. Key skills include experience with performance tuning, distributed systems, and machine learning serving, particularly in areas like GPU utilization and batching. This position is ideal for seasoned engineers with a strong track record in production systems who thrive on solving complex challenges. Chai Discovery fosters a culture of ownership and rapid innovation, making it a dynamic environment for technical excellence.

Joblaze insights

  • Listed about a month ago — first seen on Joblaze September 3, 2026. Last confirmed on Chai Discovery's careers page October 31, 2026.
  • AI/ML appears in 42.5% of 1069 comparable senior ai/ml roles; GPU appears in 2.3% of 1069 comparable senior ai/ml roles.

Quick facts

How much experience is required?
At least 4 years of relevant experience for this Software Engineer, Compute Infrastructure role.
What's the tech stack?
Joblaze extracted these technologies from the posting: AI/ML, GPU, ML serving, distributed systems.
What seniority level is this role?
Chai Discovery targets senior candidates for this position.
Is this full-time or contract?
Full-time for this Software Engineer, Compute Infrastructure role at Chai Discovery.

From the original posting

About the role

Build the compute infrastructure behind Chai’s next generation of models, scientific products, and autonomous research. Our research program and runtime inference constantly demand more compute, and the rate at which we bring it online and manage it effectively directly inflects Chai's velocity.

  • Bring new clusters up and own the systems that make them useful across the business.

  • Work with researchers, engineers, and scientists to remove performance and reliability bottlenecks across the stack.

  • Optimize model inference and serving to run more scientific work with available compute.

  • Build the shared storage and security foundations that let people and agents use compute and data safely.

About you

We look for depth in distributed systems, compute infrastructure, and performance engineering, with demonstrated ownership of substantial production systems. You can reason across layers and make sound tradeoffs between complexity, reliability, and efficiency.

Standard company text repeated across Chai Discovery's postings is omitted here.

Similar positions

Chai Discovery
Software Engineer, AI Platform
Chai Discovery · San Francisco office
Chai Discovery
Research Engineer, ML Infrastructure
Chai Discovery · San Francisco office
Chai Discovery
Software Engineer, Developer & Research Platform
Chai Discovery · San Francisco office
Chai Discovery
Research Engineer
Chai Discovery · San Francisco office
Chai Discovery
Software Engineer, Scientific Products
Chai Discovery · San Francisco office