Staff/Sr Staff AI Research Engineer

About Xaira Therapeutics
Xaira Therapeutics is an integrated biotechnology company driving advances in artificial intelligence to learn the language of life and transform how we treat disease. The company seeks to rethink the drug discovery and development process from end-to-end by bringing together leading talent across three core areas: machine learning research to better understand biology, expansive data generation to power new models, and robust therapeutic product development to treat disease. Xaira is headquartered in the San Francisco Bay Area.
Position Overview

We are seeking an experienced AI Research Engineer to join our team and lead some of our efforts in building the AI powered future of drug discovery. As a senior member of our interdisciplinary group, you will collaborate closely with AI scientists, other engineers, and drug discovery experts to help develop AI models to accelerate the identification and development of novel therapeutics. This role offers a unique opportunity to leverage cutting-edge AI techniques to address pressing challenges in healthcare and contribute to improving human health globally.

Key Responsibilities

• Drive the development of state-of-the-art AI models for drug discovery tasks, leveraging deep learning frameworks.
• Scale distributed training of large AI models on GPUs.
• Profile and optimize AI models in training and inference contexts.
• Stay abreast of the latest advancements in AI, machine learning and proactively identify opportunities for innovation and improvement.
• Contribute to technical documentation, reports, and presentations to communicate findings, methodologies, and insights clearly and effectively.
• Provide deep technical leadership on the development and optimization of core AI models that power our drug discovery efforts. 

Minimum Qualifications

• Bachelor's degree in Computer Science, Engineering, or related fields.
• 7+ years of industry experience in AI, machine learning, or related roles, with a track record of successfully implementing and deploying AI models in production environments.
• Proficiency in programming languages such as Python, C++ and deep expertise with frameworks such as PyTorch/Jax
• Strong understanding of deep learning fundamentals, including neural network architectures, optimization techniques, and evaluation metrics.
• Experience in distributed training and inference of large scale AI models on thousands of GPUs using DDP, FSDP etc
• Experience with performance optimization of large scale AI  models
• Effective communication and collaboration abilities, with the capacity to work effectively in a fast-paced, multidisciplinary team environment.
• Experience solving complex problems and comparing alternative solutions, tradeoffs, and diverse points of view to determine a path forward.

Preferred Qualifications:

• Publication track record at conferences such as NeurIPS, ICML, ICLR, etc.
• Experience working and communicating cross-functionally on an interdisciplinary team.
• Experience with different GPU architectures, cloud environments and investigating tradeoffs
• Experience with low level hardware optimization and with custom CUDA kernels
• Contributions to open source projects

Benefits

• Opportunity to make a meaningful impact by contributing to groundbreaking research at the intersection of AI and healthcare.
• Competitive compensation package, including salary, bonuses, and benefits.
• Professional development opportunities, including access to conferences, workshops, and training programs.
• A collaborative and inclusive work culture that values diversity, creativity, and continuous learning.
• Chance to work with and develop cutting-edge technologies and contribute to projects with significant societal benefit.

If you are a motivated individual with a passion for applying AI to advance drug discovery and improve human health, we encourage you to apply and join us in our mission to make a positive difference in the world.

Compensation
We offer a competitive compensation and benefits package, seeking to provide an open, flexible, and friendly work environment to empower employees and provide them with a platform to develop their long-term careers. A Summary of Benefits is available for all applicants. We offer a competitive package that includes base salary, bonus, and equity. The base pay range for this position is expected to be $200,000 - $280,000 annually; however, the base pay offered may vary depending on the market, job-related knowledge, skills and capabilities, and experience.

Xaira Therapeutics an equal-opportunity employer. We believe that our strength is in our differences. Our goal to build a diverse and inclusive team began on day one, and it will never end.

TO ALL RECRUITMENT AGENCIES: Xaira Therapeutics does not accept agency resumes. Please do not forward resumes to our jobs alias or employees. Xaira Therapeutics is not responsible for any fees related to unsolicited resumes.