10 of 10 Google 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 ...

Solid grounding in machine learning/deep-learning fundamentals — you can build, train, and debug neural networks from scratch in a modern framework (PyTorch, JAX, or similar). Demonstrable hands-on experience: research projects, publications, internships, open-source, or production work. Comfortable reading ML papers and reimplementing them. ...

equivalent practical experience. Strong Python programming and software‐engineering skills. Experience developing machine‐learning, generative AI or LLM applications. Practical experience with PyTorch, JAX, TensorFlow or a similar framework. Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration. Knowledge of reinforcement learning, preference optimisation ...

Nice to have Experience with optimisation, reinforcement learning and/or large language models Familiarity with deep learning frameworks such as TensorFlow, PyTorch or JAX Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research A track record ...

Succeed Experience building, training, adapting, or evaluating machine learning models. Strong software engineering skills in Python and modern ML frameworks such as PyTorch, JAX, or TensorFlow. Experience designing experiments, evaluation metrics, or test sets for model performance. Ability to debug model behavior using data, traces, logs, and qualitative feedback. ...

About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

neuromorphic engineers, and keep models honest against field conditions. What you’ll bring Experience with ML for time-series or biosignals (PyTorch/JAX), model evaluation under noisy real-world data, and a bias toward deployable systems. Apply for Machine Learning Engineer, Neural Decoding — Manchester/Los Angeles #J ...

ensuring models stay honest against field conditions and remain deployable in real-time systems. Experience with time-series or biosignals ML using PyTorch or JAX is highly valued, with hybrid work across Manchester and Los Angeles. #J-18808-Ljbffr ...

Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well as from collaborations with research programs at partner universities and technical institutes all over the world. From … active contributor to the wider research community by partnering with universities and publishing papers.Artificial intelligence will be one of humanitys most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

experience. We do not expect applicants to be experts in all fields simultaneously. Artificial intelligence will be one of humanitys most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high … years of experience working in frontier AI research labs on pre-training or post-training teams. Experience writing TPU/GPU kernels (e.g., JAX, CUDA) to optimize model performance and real-time inference. Experience designing, training, and scaling generative pixel or real-time video architectures. Track record of cross ...