This list contains only the countries for which job offers have been published in the selected language (e.g., in the French version, only job offers written in French are displayed, and in the English version, only those in English).
We are seeking an AI Research Engineer to help design, train, evaluate, and optimize Chai’s core models and infrastructure, working in tight loops with research scientists. 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 craft-obsessed team of micro-pessimists and macro-optimists.
Job Responsibility:
Develop and optimize all the infrastructure & tooling upon which our AI research depends
Collaborate with AI researchers to understand, implement and optimize new research directions
Integrate AI models into production systems, ensuring scalability and reliability
Requirements:
Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field
Strong proficiency in programming languages such as Python and deep learning frameworks like PyTorch
Deep understanding of operating system internals
Experience with HPC infrastructure such as Slurm and Kubernetes
Experience with performance engineering and profiling for CPU and CPU workloads
Experience writing, building, and maintaining large-scale ETL pipelines
Nice to have:
Ideal backgrounds include high-performance computing, custom CUDA kernels and GPU programming, AI/ML + data systems at scale
What we offer:
Frontier technology: work on the technologies that promises to fundamentally change how new therapeutics are created
World-class people: work with world-renowned leaders in the AI x Biology ecosystem
Velocity and ownership: we move incredibly fast and liberally distribute ownership, which means continuous learning and career growth opportunities are guaranteed
Strong culture and mission: we foster a supportive culture that is mission-driven and high-energy
Highly competitive compensation: we attract the best talent and compensate them accordingly