3 of 3 Remote/Hybrid Google JAX Jobs in London

modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training ...

classifiers Behaviour-shaping or reward-modelling components. LLM and multimodal fine-tunes Adversarial robustness defences Use Python and modern ML frameworks (e.g., PyTorch, TensorFlow, JAX, HuggingFace). Contribute to creation of synthetic datasets, adversarial evaluation corpora, and scenario-based test sets. Help transition research outputs into scalable controls … Responsible AI concepts: Bias and fairness Explainability Privacy-preserving ML. Robustness and uncertainty Nice to have: Experience with ML frameworks: PyTorch, TensorFlow, Flax/JAX, HuggingFace. Exposure to multimodal models (CLIP, Whisper, LLaVA, video transformers). Familiarity with safety benchmarks, adversarial testing, red teaming, or uncertainty estimation. Knowledge ...

Deep Learning Engineer About the Company Join a fast-growing AI-driven technology company that’s modernising prediction and decision systems for complex, real-world industries. The organisation builds powerful AI platforms that help partners ...