1 of 1 Contract Google JAX Jobs in the UK

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 ...