12 of 12 Google JAX Jobs in the UK excluding London

quantitative discipline. Research experience involving machine learning, deep learning, optimisation or statistical modelling. Strong Python programming skills. Experience with frameworks such as PyTorch, TensorFlow, JAX or Keras. Experience designing experiments, evaluating model performance and interpreting results. A curious mindset and a passion for exploring emerging technologies. Excellent written ...

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

/or large language models Experience in chip design and applied AI research Familiarity with deep learning frameworks such as TensorFlow, PyTorch or JAX A track record of securing research funding and building successful collaborations If this sounds interesting and you'd like to learn more, click the link ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research ...

equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field. Desirable: Exposure to machine learning frameworks such as PyTorch, JAX, Triton, TensorFlow Experience with distributed workload management systems such as Kubernetes, VLLM, Keras or MLOps pipelines Experience working with hardware simulators or emulators (e.g. QEMU). ...

others); OR equivalent work experience delivering state-of-the-art AI/ML solutions. Experience with at least one major deep learning framework (PyTorch, JAX, TensorFlow). Familiarity with machine learning literature and state-of-the-art approaches. Experience developing and delivering robust software solutions, including demonstrated advanced programming ...

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile ...

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

Lead AI Engineer Early employee at a high-growth AI startup Ideal Experience Strong Python software engineering skills Expertise with PyTorch and/or JAX Experience deploying machine learning systems into production Deep knowledge of LLMs and Generative AI Hands-on experience with LoRA, QLoRA, DPO, RLHF ...

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc click apply for full job details ...

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobil... WKCL1_UKTJ ...

We are supporting a pioneering technology organisation seeking a Senior Research Engineer to lead complex AI research and engineering initiatives. This position sits at the intersection of advanced research and practical engineering. You will help ...