26 to 50 of 70 Google JAX Jobs in London

training RL agents and iterating on reward design, with strong intuitions for what works and why Proficiency in Python and ML frameworks (PyTorch or JAX) with the ability to implement and run experiments independently Track record of research output: publications, open-source contributions, or impactful technical projects Willingness ...

related Expertise in generative modelling, machine learning, music information retrieval or speech and audio processing Industry experience of coding in Python and PyTorch or JAX Excellent communication and presentation skills Preferred Qualifications Deep expertise in one of the following areas: audio generation and editing, post-training of diffusion models ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

training techniques — architecture, tokenization, fine-tuning, and evaluation. Hands-on experience with Python and PyTorch; you write clean, experimental code and iterate fast. JAX is acceptable with strong Python proficiency. C++ and Rust are not required, CUDA and GPU knowledge is expected but hands-on experience is not required. ...

with evaluating and improving the generalizability and robustness of ML systems. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross‐functional teams and present findings. One or more papers at top‐tier venues (such ...

YSDA, or international equivalents). Strong foundation in probability, statistics, linear algebra, and algorithms. Tech Stack & Hard Skills: Python (NumPy, Numba). PyTorch or JAX (for the ML track) OR advanced backtesting/quantitative modeling frameworks (for the Quant track). Proven track record of developing profitable ML models ...

creative teams to turn ML capabilities into compelling experiences What We're Looking For: Strong knowledge of Python Experience writing models in PyTorch or JAX A good understanding of how large language models work Comfortable working along the whole ML pipeline: from data preparation to model design, training ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. Familiarity with modern software engineering practices. You document experiments clearly and communicate trade‐offs ...

obvious solutions. Qualifications Strong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experience Must have two or more years work experience building deep learning systems, for any domain: robotics ...

tools, frameworks, and architectural patterns Qualifications PhD in CS/CE/EE, or equivalent, in industry experience Deep knowledge of PyTorch, TrensorFlow or Jax Knowledge of model training framework In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models Experience ...

tools, frameworks, and architectural patterns Qualifications PhD in CS/CE/EE, or equivalent, in industry experience Deep knowledge of PyTorch, TrensorFlow or Jax Knowledge of model training framework In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models Experience ...

experience with other robotics simulators (Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, Brax). PyTorch, our user-facing interface. Other JIT compilers such as Triton, JAX, or Numba. 3D tooling and interactive editors (Maya, Blender, Houdini). Serialization formats (USD, glTF, MJCF/URDF). C++, CUDA, or Metal. Developer ...

Share Senior Software Engineer, TorchTPU Google London, UK X In most instances, this position requires in-person interviews as part of the hiring process. Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience. 5 years of experience with software development in one or more … software oriented projects from inception to delivery. Preferred qualifications: 5 years of experience with data structures and algorithms. Experience with ML frameworks (e.g., TensorFlow, JAX, PyTorch) and hardware performance analysis, particularly with TPUs and GPUs. Experience with Compilers and Compiler construction. Experience programming in Python. Ability to effectively navigate ...

Build large-scale, observable ML systems. Work with C, C++, Python, CUDA, and other low-level GPU technologies. Python and/or C++. PyTorch, JAX, TensorFlow, or another deep-learning library. GPU programming with CUDA, Triton, SYCL, ROCm, or similar tools. Large-scale ML systems, including very large training ...

image/video generative models Experience applying reinforcement learning with deep neural networks in practical domains Strong proficiency in Python and PyTorch or JAX, including debugging and profiling numerical code Self-driven with clear communication, documentation, and concise presentation of trade-offs Desirable: experience with robotics simulators (e.g., Isaac ...

oral communication skills. Preferred Qualifications GPU compute programming models & optimization techniques. Machine learning development using one or more ML frameworks (PyTorch, TensorFlow. or JAX). Experience with system level programming and computer architecture. At Apple, we're not all the same. And that's our greatest strength. We draw ...

models — architecture, training, and inference. Experience solving real problems using reinforcement learning with deep neural networks in any domain. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You are self‐driven, pro‐active, communicate efficiently, document experiments clearly and communicate trade ...

small datasets. Clear writing and communication about modelling choices, results and their limitations, for both ML and scientific audiences. Strong Python and PyTorch or JAX, and daily use of coding agents. NICE TO HAVE Experience post-training LLM agents or protein models with reinforcement learning, for example with ProtRL. ...

develop new tools that are designed to be functional and easy to learn and use. Key Requirements Experience with inference frameworks such as JAX and/or PyTorch Experience of software development in Python, Rust, C/C++, or a similar language Experience monitoring and optimising code written ...

performance, reliability, and operation. Strong Python and generalist software engineering skills across backend services, data systems, and internal tools, with familiarity with PyTorch, JAX, or similar ML frameworks. Experience with distributed training or inference, including diagnosing performance bottlenecks and failures across the stack. Understanding of post-training workflows; experience ...

develop new tools that are designed to be functional and easy to learn and use. Key Requirements Experience with inference frameworks such as JAX and/or PyTorch Experience of software development in Python, Rust, C/C++, or a similar language Experience monitoring and optimising code written ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command-line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases — schema design, query performance, and day-to-day operational management. Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command-line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases - schema design, query performance, and day-to-day operational management. Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

team pushing real-time speech technology. Ideal candidates have MSc/PhD in ML or related fields and hands-on experience with PyTorch/JAX/TF, plus several years in research or industry roles. #J-18808-Ljbffr ...

significant experience designing and building production-grade machine learning systems at scale. You have deep experience with modern ML frameworks such as PyTorch, TensorFlow, JAX, or similar. You have experience with—or a strong interest in—multimodal machine learning across areas such as text, audio, image, or video. ...