Level

Chai Discovery

Research Engineer - Auto Research

AI in this role

pytorchjax
ai-research

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

Build agent-powered systems and frameworks to improve model training, accelerate inference, and help our research organization move faster. Automate the research loop itself rather than just the models it produces, including the tooling and datasets that allow agents to run experiments, evaluate results and propose next steps.

Chai's models are moving beyond protein structure prediction into real-world therapeutic engineering. This is a chance to push the frontier of AI drug design, working alongside a rigorous and craft-obsessed team.

About you

Ideal backgrounds include deep industry experience working with top AI/ML teams on agent-powered software—with strong system design skills, proficiency in Python, and Pytorch or JAX fluency. We look for technical spikes where you have gone deep and demonstrated exceptional impact on real-world problems and systems.

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.

How we score this

Research Engineer - Auto Research at Chai Discovery scores 91 out of 100 on AI centrality, which makes it AI Level 4 of 4 (Builds AI) on this board. The level measures how much of the work is AI, not seniority.

Classification

AI Level 4. Building AI systems is the job itself: without AI, the role would not exist.

  1. AI Level 480 to 100
  2. AI Level 360 to 79
  3. AI Level 240 to 59
  4. AI Level 10 to 39

Bands come from how often the tools, models and workflows of the role are named in the posting itself. Open the description and count.

Prepare for this job

A free preview built only from this posting: what it asks for, what you could be asked in an interview, and how to adjust your resume.

Skills and AI tools this role asks for

AI ResearchPyTorchJax

Questions you could be asked

  1. Tell me about a research question you investigated. What did you find?
  2. Walk me through how you've used PyTorch in your day-to-day work.
  3. What are the limits of Jax that you've run into, and how did you work around them?
  4. How would you decide a model or AI system is ready to ship?
  5. Tell me about a time a model underperformed in production. How did you find out, and what did you change?

Adapt your resume

  • List these exact terms on your resume: AI Research, PyTorch, and Jax. An applicant tracking system matches the wording, not the idea.
  • Attach one line of real, concrete experience to at least one of them — a tool named with nothing behind it rarely survives a human read.
  • Lead with what you built, trained or shipped — this role is judged on the AI system itself, not the tools around it.

Want your resume actually rewritten for this job?

The free preview above is everything we have today. A full resume rewrite is not live yet and has no price set. Join the waitlist and we will email you if we open it.

Get new AI jobs at AI Level 4+ by email

One email a week with the new AI jobs at AI Level 4+, each rated AI Level 1 to 4 for how much AI is in the work. No recruiter spam, unsubscribe in one click.

Free. One email a week. Unsubscribe in one click.

Similar roles

Research roles rated AI Level 4 at other companies.

What kind of AI work fits you?

Answer 12 practical questions in about three minutes. Get a simple profile, the work it points to, and live roles to explore next.

Find my next step

More jobs at Chai Discovery

Related searches

Same AI level