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

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

networks, training vs inference, numerical precision, performance trade‐offs). Hands‐on experience with one or more major ML frameworks such as PyTorch, TensorFlow, JAX, or similar. Strong proficiency in Python for ML workflows, experimentation, and automation. Experience designing, running, and analysing ML benchmarks or experiments. Experience working ...

least two of the following areas: Multimodal Learning Computer Vision Natural Language Processing Recommendation Systems Representation Learning Hands‐on experience with PyTorch, TensorFlow, or JAX Experience working with Large Language Models, Vision Language Models, or Multimodal LLMs Experience deploying machine learning models into large-scale production environments By applying ...

improve the Waymo Driver. Requirements include a PhD in ML or related field (or MSc with publications), strong Python, and 3+ years with JAX, TensorFlow or PyTorch. C++ is a plus. #J-18808-Ljbffr ...

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

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

Team The frameworks team is responsible for ensuring Graphcore hardware enjoys seamless support in user-facing machine learning frameworks such as Pytorch, Triton, Jax and TensorFlow. We have strong focus on both ease-of-use and performance. Our software team is at the cutting edge of the machine learning ...

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

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver—to improve access to mobility … with a strong academic publication record Strong statistical/ML knowledge Proficiency in Python 3+ years of experience with modern deep learning frameworks: JAX, TensorFlow/PyTorch We prefer: Proven track record of deploying ML models to production environments Deep learning experience, especially with generative models, e.g., LLMs/ ...