• Companies
  • News
By technology
  • Python10.5k
  • AWS7k
  • Kubernetes5k
  • Docker4.5k
  • PostgreSQL3.7k
  • TypeScript3.1k
  • SQL2.9k
By category
  • Backend Developer10.6k
  • DevOps Engineer3k
  • Machine Learning Engineer2.8k
  • Data Engineer2.4k
  • Full Stack Developer2.4k
  • Embedded Engineer2k
  • Mobile Developer744
By city
  • San Francisco, CA5.2k
  • New York, NY4.6k
  • Seattle, WA1.1k
  • Toronto, ON827
  • Washington, DC827
  • Boston, MA743
  • Austin, TX729
Work mode
  • On-site18.5k
  • Remote7.9k
  • Hybrid4.6k
Job type
  • Full-time30.1k
  • Part-time440
  • Contract406
  • Freelance145
  • internship4
  • seasonal1
Experience
  • Entry level
  • Mid level
  • Senior
  • Staff+
  • Engineering management

The job board for developers. Find your next role at companies that value great engineering.

© 2023-2026 . All rights reserved.

Logos provided by Logo.dev
All jobs

Software Engineer, AI Platform

Chai Discovery · San Francisco office

Extracted from the posting

Base
Not stated
Remote
On-site
Stack
JAX, Python, PyTorch, AlphaFold (company stack)

Build the AI platform that lets scientists direct drug discovery research programs with agents, using Chai's foundation models to design molecules and guide experiments. The work includes core abstractions for agent execution, model inference, and scientific tools, plus products that shorten the cycle from idea to experiment to decision. The posting does not name a specific tech stack.

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 AI platform that lets scientists direct research programs with agents. We believe that as Chai's models get more powerful, scientists will focus more on fundamental problems and biological hypotheses, and delegate more of the subtasks to answer them to agents. Making this possible requires inventing the abstractions through which people, agents, and Chai’s models collaborate.

  • Work with scientists and model researchers to define how agents use Chai’s foundation models to design molecules and guide experiments.
  • Build products that shorten the cycle from idea ➝ to experiment ➝ to the next decision, while scientists monitor and direct the work.
  • Build the core abstractions for agent execution, model inference, and scientific tools so new capabilities come together fluidly across the workspace.


About you

We look for technical depth, strong software architecture, and product judgment, with experience building complex applications or agent systems. You can find clear abstractions in a messy domain and build systems that hold up as their scope grows. We value a practical understanding of where agents fail and how to tell whether they’re doing useful work.


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.

Company stack

This posting doesn’t list technologies. These are the ones Chai Discovery uses across its other roles.

JAXPythonPyTorchAlphaFoldESMProteinMPNNRFDiffusion
Chai Discovery
Chai Discovery
Website

Job details

TypeFull-time
Work ModeOn-site
LevelMid Level
Location🇺🇸San Francisco, CA
PostedNov 05, 2026
Apply Now

Related jobs

GreenLite

AI Engineer

GreenLite
🇺🇸 New York, NYHybrid · Full-timePython · TypeScript · LangGraph
Zscaler

Principal Software Engineer - Intelligent Data Platform

Zscaler
$185,500 - $265,000🇺🇸 Santa Clara, CAHybrid · Full-timeJava · Go · Python
Formal

Staff Software Engineer | Systems and L7 Protocols

Formal
🇺🇸 San Francisco, CAOn-site · Full-timeGo · Rust · Temporal