Senior/ Staff Research Engineer

Senior/Staff AI Research Engineer Location: London (Hybrid) Salary: £150-250k base generous equity Are you excited to drive AI drug discovery forward by scaling SOTA models and optimizing neural nets on GPUs? Qualifications • Bachelor's/ Master's or PhD in Computer Science, Engineering, or a related field • 5 years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level hardware tuning and custom CUDA kernel development Key Responsibilities • Develop cutting-edge AI models for drug discovery using deep learning frameworks • Scale and optimize distributed training of large AI models on GPUs • Enhance performance of AI models during both training and inference stages • Communicate findings through documentation, presentations, and stay updated on AI advancements Benefits • Generous sock options • Competitive base • Private healthcare • Generous pension contributions • Contribute to impactful research at the AI-healthcare frontier • Access to professional growth via conferences, workshops, and training • Inclusive, collaborative culture focused on innovation and continuous learning If you meet 60-70% of the requirements I would love to hear from you. Click the Easy Apply button!