14 of 14 Google JAX Jobs

language processing, including hands-on experience with transformer architectures (e.g., GPT, BERT, T5). Expertise in deep learning frameworks such as TensorFlow , PyTorch , or JAX . Strong knowledge of machine learning algorithms and their application to NLP tasks. Experience with cloud platforms such as AWS, GCP, or Azure ...

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) Experience influencing organizational policy and securing major research funding ...

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

boundaries of what AI can actually do. Experience required: Strong hands-on experience with Reinforcement Learning (RL) Python and modern ML frameworks ( PyTorch, JAX, TensorFlow ) Experience working with time-series or sensor data Ability to turn real-world problems into practical ML solutions Comfortable taking models from research into ...

reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor ...

message types, and topic debugging* Experience working with physical robot hardware, camera streams, joint states, or latency-sensitive control paths* OpenCV, NumPy, pandas, PyTorch, JAX, cloud, HPC, or LeRobot dataset format experience* Experience working closely with ML researchers or robotics engineers in a fast-moving technical environment The Company ...

knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science ...

work in bwHPC-S5. Ideally, your profile is supplemented by: Experience with AI frameworks, workloads (e.g. Natural Language Processing), and libraries (e.g., PyTorch or JAX). Experience with parallel programming paradigms. Some background in the field of (green) transport (e.g., fleet optimisation, energy-efficient logistics, smart city planning). ...

learning systems, with shipped models or published research. Hands-on experience with Vision Language Models, LLMs, or generative models. Strong Python and PyTorch/JAX skills, including debugging, profiling, and shipping. A passion for bridging the gap between simulation and reality. This is a technically challenging role with ...

work What you bring Engineering/physics degree Strong RL experience (deep RL, debugging, non‐trivial problems) Python + modern ML stack (PyTorch/JAX, NumPy, RL libs) Comfortable with time‐series sensor data Ability to turn ambiguous operational challenges into tractable ML problems Happy switching between research ...

3. A working knowledge and practical experience with Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL. Key Words: Machine Learning Engineer/Data Processing/Model Building/NMR/Deep Learning/Computational Biophysics/ ...

outcome.World Wide Technology UK is looking for a hands-on Cloud Engineer with strong expertise in HPC and AI/ML performance workloads on Google Cloud Platform (GCP). The role focuses on benchmarking, optimizing, and validating performance across advanced accelerator platforms including NVIDIA GPUs, AMD GPUs, and Google … inference, scientific workloads)Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods)Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm)Identify system bottlenecks (compute, memory, network, I/O)Build automation tools for benchmarking and reportingCollaborate with teams to align workloads with ...

World Wide Technology UK is looking for a hands-on Cloud Engineer with strong expertise in HPC and AI/ML performance workloads on Google Cloud Platform (GCP). The role focuses on benchmarking, optimizing, and validating performance across advanced accelerator platforms including NVIDIA GPUs, AMD GPUs, and Google … inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads ...

World Wide Technology UK is looking for a hands-on Cloud Engineer with strong expertise in HPC and AI/ML performance workloads on Google Cloud Platform (GCP). The role focuses on benchmarking, optimizing, and validating performance across advanced accelerator platforms including NVIDIA GPUs, AMD GPUs, and Google … inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads ...