8 of 8 Google JAX Jobs in the East of England

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

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

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