Research Engineer (LLM Performance), London

Onsite London

gcppytorch ml research

greenhouse

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

Join a team focused on enhancing AI models for drug design in London. You'll work on optimizing performance and collaborating with scientists to advance biopharmaceutical research.

Details

London, work from office
Significant experience with large scale distributed training of LLMs
Experience with deep learning ML frameworks (either JAX or PyTorch)
Good understanding of GPU architectures

The work

Implement and optimize LLM post-training methods at scale
Collaborate with research teams to translate new methods into production-ready systems
Diagnose and fix performance bottlenecks in distributed training and inference systems
Deploy low-precision methods to balance performance with accuracy