2 of 2 Remote/Hybrid JAX Jobs in the UK excluding London

Senior) or 8+ years (Staff) of hands‐on applied ML experience, including production‐grade work. Deep familiarity with modern deep learning stack (PyTorch or JAX), and breadth across at least two of: generative models (diffusion, autoregressive, flow‐based), graph and equivariant networks, vision (CNNs, ViTs, multimodal LLMs), search and planning ...

shape what responsible AI security looks like in practice About you (skills/experience): You will bring A genuine understanding of deep learning (JAX, PyTorch, or TensorFlow), and you've worked with large-scale cloud training or inference infrastructure Think like an attacker on AI systems: prompt injection, model inversion ...