Chai Discovery Logo

Chai Discovery

Software Engineer, Compute Infrastructure

Reposted 2 Months Ago
In-Office
San Francisco, CA, USA
Senior level
In-Office
San Francisco, CA, USA
Senior level
We are seeking a Software Engineer focused on building resilient infrastructure systems for AI drug discovery, enhancing developer productivity and ensuring compliance with biopharma standards.
The summary above was generated by AI
About Chai Discovery

Chai Discovery builds the computer-aided design suite for molecules. Eli Lilly, GSK, Pfizer, Bristol Myers Squibb, argenx and Novartis all use Chai to power drug discovery programs. AI is reinventing life sciences the same way it reinvented software engineering, and our team is building that future.

We train frontier models that learn the underlying foundations of biochemical structure and interaction. We pair this with a computational workspace to iterate on molecules from idea to medicine, helping scientists move faster and develop medicines beyond the reach of existing methods.

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

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.

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 and promote a culture of high velocity and ownership. We compensate our team accordingly.

HQ

Chai Discovery San Francisco, California, USA Office

San Francisco, California, United States

Similar Jobs

One Month Ago
In-Office
Mountain View, CA, USA
164K-205K Annually
Senior level
164K-205K Annually
Senior level
Big Data • Machine Learning • Software • Analytics • Big Data Analytics
Design and build Databricks’ multi-cloud compute infrastructure and serverless platform. Develop highly available, low-latency systems supporting heterogeneous workloads, GPU qualification, stateful execution, and scale-to-zero efficiency. Build cloud abstractions across AWS, Azure, and GCP while optimizing API gateways, networking, container orchestration, and distributed systems at scale.
Top Skills: AWSAzureC++DockerFirecrackerGCPJavaKubernetesMicrovmPythonScala
2 Months Ago
In-Office
San Francisco, CA, USA
240K-280K Annually
Senior level
240K-280K Annually
Senior level
Artificial Intelligence • Information Technology
Design and implement manifest-driven provisioning and reconciliation systems for GPU hosts and clusters. Build durable workflow engines, declarative self-service APIs, event-driven automation, self-healing pipelines, and production-grade control planes. Own coding, testing, CI/CD, and operate the platform in production to eliminate manual provisioning.
Top Skills: BgpBmcCadenceCi/CdCudaGoInfinibandIpmiIpxeKafkaKubernetesKubernetes ControllersNatsNcclPxePythonRedfishRoceRustSqsTemporalVlan
2 Months Ago
Hybrid
San Francisco, CA, USA
230K-405K Annually
Mid level
230K-405K Annually
Mid level
Artificial Intelligence • Machine Learning • Generative AI
Develop and optimize large-scale compute systems for AI workloads, improving infrastructure reliability and performance across various layers of the stack.
Top Skills: Agent InfrastructureBenchmarkingCaasDistributed SystemsGpu InfrastructureHardware-Aware Performance OptimizationHigh-Performance ComputingKubernetesNcclNetworking ProtocolsObservabilityRdmaStorage

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account