4 of 4 Google JAX Jobs in the UK excluding London

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

others); OR equivalent work experience delivering state-of-the-art AI/ML solutions. Experience with at least one major deep learning framework (PyTorch, JAX, TensorFlow). Familiarity with machine learning literature and state-of-the-art approaches. Experience developing and delivering robust software solutions, including demonstrated advanced programming ...

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

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