Senior/ Staff Research Engineer

JR United Kingdom; Slough; On-site; GBP 70,000 - 100,000; Full time

Job summary

A leading company is seeking an experienced professional to drive AI drug discovery initiatives, focusing on developing and optimizing AI models using state-of-the-art technology. This role offers opportunities for professional growth in an inclusive and innovative environment, ideal for those passionate about advancing AI in healthcare.

Benefits

Access to conferences and training

Collaborative culture focused on innovation

Qualifications

• 5+ years of experience deploying AI/ML models in production settings.
• Skilled in model performance optimization and low-level hardware tuning.
• Familiar with cloud platforms and hardware tradeoffs.

Responsibilities

• Develop AI models for drug discovery using deep learning frameworks.
• Optimize distributed training of large AI models on GPUs.
• Communicate findings through documentation and presentations.

Skills

Python

C++

Distributed Training

Deep Learning

GPU Architectures

Education

Bachelor's/Master's/PhD in Computer Science, Engineering, or related field

Tools

PyTorch

Jax

CUDA

Job description

Social network you want to login/join with:

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

• 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!