1 of 1 Remote/Hybrid Google JAX Jobs in London

Research Engineer (LLM Performance), London

Hiring Organisation
Isomorphic Labs
Location
London, UK
Employment Type
Full-time
world drug design programs. Skills and qualifications Essential: Significant experience with large scale distributed training of LLMs. Experience with deep learning ML frameworks (either JAX or PyTorch).Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).Good understanding of GPU architectures. Reasoning about performance concepts is more important ...