Staff/Sr Staff AI Research Engineer

Company: Tbwa Chiat/Day Inc
Location: Seattle
Posted on: February 1, 2025

Job Description:

Xaira is an innovative biotech startup focused on leveraging AI to transform drug discovery and development. The company is leading the development of generative AI models to design protein and antibody therapeutics, enabling the creation of medicines against historically hard-to-drug molecular targets. It is also developing foundation models for biology and disease to enable better target elucidation and patient stratification. Collectively, these technologies aim to continually enable the identification of novel therapies and to improve success in drug development. Xaira is headquartered in the San Francisco Bay Area, Seattle, and London.About Xaira TherapeuticsXaira 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 OverviewWe 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.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.CompensationWe 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. 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 is 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.
      #J-18808-Ljbffr

Keywords: Tbwa Chiat/Day Inc, Edmonds , Staff/Sr Staff AI Research Engineer, Engineering , Seattle, Washington