1 to 25 of 217 Google JAX Jobs in the UK

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 More ❯

Biases, Kubeflow, Seldon Core Data Processing: Pandas, NumPy, Apache Spark, Beam Model Serving: TensorFlow Serving, TorchServe, FastAPI, Flask Cloud Platforms: AWS (SageMaker, S3, EC2), Google Cloud AI Platform, Azure ML Orchestration: Docker, Kubernetes, Airflow Databases: PostgreSQL, BigQuery, MongoDB, Redis Experiment Tracking & Monitoring: MLflow, Neptune.ai, Weights & Biases Version Control: Git More ❯

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 More ❯

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 More ❯

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 More ❯

such as computational pathology, genomics, transcriptomics or medical imaging. Strong experience with Python and at least one deep learning frameworks such as PyTorch, TensorFlow, JAX, and PyTorch Lightning. Familiarity with packages and technologies such as NumPy, Pandas, Scikit-learn, Scikit-image, OpenCV, Git, and Bash. Experience working with HPC More ❯

versioning, testing, CI/CD, API design, MLOps) Building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., PyTorch, MLFlow, JAX) Distributed computing frameworks (e.g., Spark, Dask) Cloud platforms (e.g., AWS, Azure, GCP) and HP computing Containerization and orchestration (Docker, Kubernetes) Ability to scope and More ❯

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 More ❯

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 More ❯

of relevant working experience - Experience with machine learning fundamentals, with working knowledge of Python and experience with deep learning frameworks such as Pytorch, TensorFlow, JAX or MXNet - 3+ years of relevant experience in developing and deploying large scale machine learning or deep learning models and/or systems into More ❯

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 More ❯

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 More ❯

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 More ❯

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 More ❯

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 More ❯

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 More ❯

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 More ❯

learning, and/or large language models (LLMs) •Experience in chip design and applied AI research •Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Why Join Us •Work on groundbreaking AI research with real-world applications and a strong commitment to open-source contributions and scientific advancement •Collaborate More ❯

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 More ❯

MLOps practices and tools including version control, automated testing, and CI/CD. Experience in at least one ML framework (PyTorch/TensorFlow/Jax) and robust experience in the Python data science ecosystem. Experience with large language models (e.g. autoregressive LLMs) for biological sequences is a plus. Familiarity More ❯

regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications: 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language More ❯

data-driven role, with exposure to: 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; developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); iterating on network architectures and More ❯

Java, and their associated security considerations. Coding experience with AI & ML, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn. Personal Attributes Drive and lead continuous improvement with energy and passion Displays high levels of self-awareness and emotional intelligence in More ❯

Staff Software Engineer, GPU Performance, Core ML link Copy link corporate_fare Google place London, UK Advanced Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders;deep expertise in domain. Apply link Copy link Bachelor's degree or equivalent practical experience. 8 years of experience in … and performance bottlenecks. Ability to develop and utilize sophisticated performance models and benchmarks to guide optimization efforts and hardware roadmap decisions. About the job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our … and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to More ❯

e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯