2 of 2 Google JAX Jobs in Scotland

hardware-aware algorithms for efficient AI acceleration Reading and implementing ideas from current AI and machine-learning research Relevant experience may include Python, PyTorch, JAX, TensorFlow, Hugging Face, LangGraph, vector databases, reinforcement learning, optimisation or distributed inference. 2. AI and Machine-Learning Algorithms You may work on: Designing ...

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