Research Engineer

Menlo Ventures; San Francisco (CA); On-site; USD 130,000 - 180,000; Full time

Benefits offered by this job

Highly competitive salary and equity package

Comprehensive healthcare and wellness benefits

Continuous learning and career growth opportunities

Collaborative and supportive startup culture

Job summary

A pioneering AI-driven drug discovery firm is seeking an AI Engineer in San Francisco. The role involves developing AI foundation models for drug discovery and requires strong programming skills, proficiency in Python, and an understanding of HPC infrastructure. Candidates should have a degree in a relevant field and experience with deep learning libraries like PyTorch. The position offers competitive salary, equity, and comprehensive benefits, along with opportunities for career growth in a collaborative startup culture.

Qualifications

• Degree in Computer Science, Software Engineering, or related field is required.
• Proficiency in programming languages such as Python is essential.
• Experience with HPC infrastructure and performance engineering is necessary.
• Experience with HPC infra (Slurm, Kubernetes)
• Experience in performance engineering (profiling, CPU/GPU optimizing, CUDA/Triton kernels)
• Experience building and maintaining large-scale ETL pipelines
• Experience with PyTorch for deep learning workloads

Responsibilities

• Collaborate with AI researchers on software systems for drug discovery applications.
• Develop and optimize HPC infrastructure.
• Integrate AI models into platforms ensuring scalability and reliability.

Skills

Python

Unix operating systems

High-performance computing (HPC)

Performance engineering

PyTorch

ETL pipelines

Performance engineering

Education

Bachelor’s or Master’s in Computer Science or related field

Tools

Slurm

Kubernetes

CUDA

Triton

PyTorch

Job description

About Chai Discovery

Chai Discovery is unlocking new biology with artificial intelligence by building state-of-the-art foundation models. The company is operating in stealth mode by a highly experienced founding team, who are well known in the AI community for seminal research accomplishments at top AI labs, have run AI departments at top AI x Biology organizations, have exited a past company via IPO, and sold AI technologies to multiple top 10 pharma companies. The company is backed with significant capital by top-tier investors.

Position Overview

Chai Discovery is seeking an AI Engineer to play a crucial role in developing our next-generation AI x biology foundation models and application to real-world drug discovery challenges. The AI Engineer will work closely with the founding team to build, optimize, and maintain the software and infrastructure that power our AI-driven drug discovery pipeline. This position is based full-time in San Francisco, CA.

Key Responsibilities

• Collaborate with AI researchers to design, implement, and maintain software systems that support AI-driven drug discovery applications.

• Develop and optimize high-performance computing (HPC) infrastructure.

• Integrate AI models into production systems, platforms, and front-end applications, ensuring scalability and reliability.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field.

• Strong proficiency in programming languages such as Python.

• Deep understanding of operating system internals (Unix).

• Experience with high-performance computing (HPC) infrastructure such as Slurm and Kubernetes.

• Experience with performance engineering (familiarity with profiling tools, optimizing CPU and GPU workloads, writing CUDA and Triton kernels, and so forth)

• Experience writing, building, and maintaining large-scale ETL pipelines.

• Experience using deep learning libraries such as PyTorch.

What We Offer

• Highly engaging work at the frontier of AI-driven drug discovery that will fundamentally change how new drugs are created.

• Opportunity to work with world-renowned leaders in the AI x biology ecosystem, contributing to impactful advancements in the field.

• Highly competitive salary and equity package.

• Comprehensive healthcare and wellness benefits.

• Continuous learning and career growth opportunities.

• Collaborative and supportive startup culture.

• A high-energy team committed to increasing the efficiency of creating new medicines and positively impacting millions of lives.

Chai Discovery, Inc. is an equal opportunity employer. We encourage applications from candidates of all backgrounds and experiences to apply.