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

Infrastructure as Code, and DevOps practices Production support, monitoring, observability, and incident management Experience with AI/ML frameworks such as PyTorch, TensorFlow, or JAX Exposure to LLMs, RAG, Agentic AI, model serving, or MLOps environments What's on Offer? Work on innovative AI and Agentic AI solutions Hybrid ...

Automation Skills Essential: Competent in writing Shell Scripts. Desirable: Experience of C, Python or Fortran programming. Experience of installing and using PyTORCH, Tensorflow, openCV & Jax Understanding the work of the Laboratory Essential: Experience working in an IT support, academic or research environment. Desirable: Research experience in Biology. People Skills ...

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

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

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

related discipline, or 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 or sequential ...

experience, with at least 2 years in an ML or AI-adjacent role Strong Python skills and familiarity with major ML frameworks (PyTorch or JAX); comfortable reading and modifying model code Hands-on experience building and operating ML pipelines in production: data pipelines, training orchestration, evaluation, and serving Experience ...

language models, retrieval systems and domain-specific tools. Qualifications include a degree in CS or related field, strong Python, and experience with PyTorch/JAX/TensorFlow; familiarity with agentic AI concepts and RL is essential. #J-18808-Ljbffr ...

silicon. What they're looking for: Degree in Physics/Engineering/Electronics and a PhD in Machine Learning Strong Python + PyTorch/JAX skills Publication record in ML or EDA venues Ability to move between research and implementation Clear communication across ML and circuit‐design audiences Experience ...

silicon. What they're looking for: First degree in relevant STEM subject 3+ years in industry building production ML systems Python + PyTorch/JAX expertise for training, inference, and model export Experience building data pipelines for complex scientific/engineering datasets Solid software engineering fundamentals - testing, CI/ ...

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

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefine what's possible ...