1 to 25 of 132 Google JAX Jobs in the UK

implement end-to-end AI pipelines: data collection/cleaning, feature engineering, model training, validation, and inference. Rapidly prototype novel models (eg, neural networks, probabilistic models) using PyTorch, TensorFlow, JAX, or equivalent. Productionize models in cloud/on-prem environments (AWS/GCP/Azure) with containerization (Docker/Kubernetes) and ensure low-latency, high-availability inference. 2. Strategic More ❯

implement end-to-end AI pipelines: data collection/cleaning, feature engineering, model training, validation, and inference. • Rapidly prototype novel models (e.g., neural networks, probabilistic models) using PyTorch, TensorFlow, JAX, or equivalent. • Productionize models in cloud/on-prem environments (AWS/GCP/Azure) with containerization (Docker/Kubernetes) and ensure low-latency, high-availability inference. 2. Strategic More ❯

implement end-to-end AI pipelines: data collection/cleaning, feature engineering, model training, validation, and inference. Rapidly prototype novel models (e.g., neural networks, probabilistic models) using PyTorch, TensorFlow, JAX, or equivalent. Productionize models in cloud/on-prem environments (AWS/GCP/Azure) with containerization (Docker/Kubernetes) and ensure low-latency, high-availability inference. Strategic Leadership More ❯

well as a strong background in AI research and applied machine learning. Expertise in key data science technologies and techniques, such as Python, Git, AWS, AWS Sagemaker, PyTorch, TensorFlow, Jax, Numpy, Scikit-learn, Time-series forecasting, Classification, Regression, Large-language models, and Experiment Design. Strong stakeholder management skills and the ability to communicate complex ideas to a diverse audience. More ❯

and technology experts. • Experience with the analysis or application of data in finance, economics, or related fields is a plus. • Experience with machine learning libraries such as PyTorch, TensorFlow, JAX is a plus. • Experience applying generative AI solutions (via open source tools or through Azure, Amazon Bedrock, Nvidia etc.) is a plus. • Experience with cloud technologies (Azure, AWS, Google More ❯

supervised models in a Bayesian Optimization context (Gaussian processes, Bayesian Neural Networks) on small and large datasets. Strong experience in at least one ML framework (PyTorch/TensorFlow/Jax) and robust experience in Python data science ecosystem (Numpy, SciPy, Pandas, etc.) Experience using a cloud computing service to reduce runtime to train and evaluate deep learning models PhD More ❯

multi-modal models and ensembles Familiar with collaborative development environments and version control systems Experience with: 2D and 3D computer graphics programming Bayesian statistics NVIDIA Warp and/or JAX, NVIDIA Modulus or similar Pytorch, Pytorch Lightning, Weights & Biases, Ray, AWS Sagemaker Experience training diverse architectures: CNNs, GNNs, Diffusion models, Contrastive learning, Transformers, LLMs, RNNs etc. at scale Computer … models and ensembles Familiar with collaborative development environments and version control systems Experience with: 2D and 3D computer graphics programming Visualization techniques Bayesian statistics NVIDIA Warp and/or JAX, NVIDIA Modulus or similar Pytorch, Pytorch Lightning, Weights & Biases, Ray, AWS Sagemaker Experience training diverse architectures: CNNs, GNNs, Diffusion models, Contrastive learning, Transformers, LLMs, RNNs etc. at scale Koopman More ❯

Python, R, C, C++, or Java, and their associated security considerations. Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. Strong understanding of More ❯

Python, R, C, C++, or Java, and their associated security considerations. Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. Strong understanding of More ❯

Python, R, C, C++, or Java, and their associated security considerations. * Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. * Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. * Strong understanding of More ❯

Python, R, C, C++, or Java, and their associated security considerations. • Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. • Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. • Strong understanding of More ❯

Python, R, C, C++, or Java, and their associated security considerations. * Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. * Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. * Strong understanding of More ❯

s or PhD degree in a quantitative field (e.g., Computer Science, Machine Learning, Statistics, Physics, Mathematics) or equivalent industry experience. Strong background in deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Experience with distributed computing platforms (AWS, GCP, Azure, or on-prem clusters). Ideal: Kubernetes and Docker experience for scalable, reproducible workflows. Working at Lila Sciences, you would More ❯

Triton); cloud computing (on hyper-scaler platforms, e.g., AWS, Azure, GCP); building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; C/C++ for computer vision, geometry processing, or scientific computing; software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

AI-driven agent systems. This role is responsible for designing scalable and intelligent cloud-native architectures that enhance personalization, product discovery, and operational efficiency for major retail customers using Google Cloud technologies. Key Responsibilities: • Architect retail-centric solutions on GCP that optimize customer experience, supply chain, and omnichannel engagement. • Implement and optimize vector search capabilities (e.g., ScaNN, FAISS) for … AI agent frameworks (e.g., Dialogflow, LangChain, or custom agent orchestration solutions). • Strong understanding of retail data ecosystems, POS integration, and customer engagement analytics. • Strong hands-on experience with Google Cloud Platform (GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g. … TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ML concepts, including deep learning, neural networks, NLP, and machine learning algorithms. • Familiarity with MLOps principles and tools for deploying, managing, and monitoring ML models in More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

Join to apply for the AI Research Scientist (AEC) role at Autodesk Continue with Google Continue with Google 2 weeks ago Be among the first 25 applicants Join to apply for the AI Research Scientist (AEC) role at Autodesk Direct message the job poster from Autodesk Principal Engineering Recruiter Product Software at Autodesk. I am an Autodesker." Job … and deploying production-ready solutions) Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry-specific data formats (e.g., BIM More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯