Senior/ Staff AI Research Engineer

The role

You will work with AI scientists, engineers and drug discovery experts to develop and deploy AI models that speed up finding new therapeutics. The work covers GPU performance, scaling model training, implementing new model architectures, and building inference tools used to design drugs.

What you will do

• Identify bottlenecks, streamline network communication, write kernels and improve hardware performance
• Help scale models, training horizons and datasets
• Implement new model architectures that translate molecules into tensors in a maintainable way
• Build inference tools used in drug design campaigns

Required qualifications

• Bachelor's degree in Computer Science, Engineering, or related fields
• Minimum of five years of industry experience in AI, machine learning, or related roles
• Track record of implementing, scaling and deploying AI models in production
• Expertise with modern deep learning toolchains and large scale model training
• Strong understanding of deep learning fundamentals
• Excellent problem-solving skills and attention to detail
• Effective communication and collaboration in a fast-paced multidisciplinary team

Nice to have

• Advanced degree
• Publications at conferences such as NeurIPS, ICML or ICLR

Benefits

• Competitive compensation and benefits package
• Open, flexible and friendly work environment
• Platform to develop long-term careers
• Summary of Benefits available to all applicants

Before you apply

• Xaira Therapeutics does not accept agency resumes

About the company

Xaira is a biotech startup using AI to transform drug discovery and development, including generative AI models for protein and antibody therapeutics. It is headquartered in the San Francisco Bay Area, Seattle and London.