Senior/ Staff AI Research Engineer

Xaira Therapeutics is seeking an experienced AI research engineer to build, scale, and deploy production machine learning systems for drug discovery. The role focuses on optimizing model training and inference, implementing architectures that incorporate biological insights, and building tools for drug design campaigns.

Responsibilities:

• GPUs go brrr. Identify bottlenecks, streamline network chatter, write kernels, and make the hardware fly
• Experiment and Scale. Help unlock bigger models, longer training horizons, larger datasets, and weirder training dynamics
• Implement the architecture. New architecture elements mix biological intuition with thinking in tensors. Translate molecules to matrices, and do it in a way that’s maintainable and clean to read
• Make some drugs. Drug design campaigns involve heavy inference workloads and the inference harness is where molecule designer needs meet ML implementation skill. Build the tools that we use to make drugs