26 to 50 of 104 Google JAX Jobs in London

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch ...

performance engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems.Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies.Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live quantitative ...

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

engineering: kernel/network tuning, lock-free programming, hardware-aware optimization, or FPGA-adjacent systems. Experience with ML/data tooling such as PyTorch, JAX, CUDA, GPU clusters, Ray, Kafka, Kubernetes, Spark, or comparable technologies. Understanding of market microstructure, backtesting pitfalls, transaction costs, and the operational needs of live ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research ...

models, including tabular foundation models. No prior familiarity with finance or financial use cases required. Preference for candidates who include a link to their Google Scholar or Semantic Scholar profile in their resume. #J-18808-Ljbffr ...

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

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

models, including tabular foundation models. No prior familiarity with finance or financial use cases required. Preference for candidates who include a link to their Google Scholar or Semantic Scholar profile in their resume. Join us to push the boundaries of AI and contribute to groundbreaking discoveries in a collaborative ...

especially knowledge in formal methods, stochastic systems and processes, the foundations of deep learning. Experience coding with deep learning libraries such as Pytorch/JAX is essential. Applicants must hold a PhD in computer science, mathematics or equivalent experience.Please see job description for a full list of requirements. What ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large‐scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large‐model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research ...

series). Experience limited to API usage or prompt engineering is not sufficient.Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks.Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking.Evidence of research/technical ...

Computer Science, Machine Learning, Physics, or a related field with a strong record of publishing novel research. Possesses hands‐on expertise in PyTorch or JAX developing modern architectures such as generative models, Bayesian optimization, or foundation models. Demonstrates experience applying machine learning to scientific domains or simulations with ...

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

About You Strong Python engineering skills and experience working in large codebases (testing, code review, CI, operational ownership). Hands‐on experience with PyTorch, JAX, or similar. Strong systems and infrastructure fundamentals. Experience with LLM training or post‐training: fine‐tuning, RL, distillation, evaluation, and/or data pipelines. ...

solid production track record Proficient in Python and at least one systems language - Rust, C++, or Go Hands-on experience with PyTorch or JAX in an industry setting Experience with inference frameworks: vLLM, SGLang, TensorRT-LLM Solid distributed systems fundamentals and experience operating production ML infrastructure Working knowledge ...

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

application is highly valued.LLM experience is valuable, but not necessary.Strong engineering skills valued, especially any of: GPU kernel development, lower-level Pytorch/JAX development, intermediate C++, FPGA/ASIC experience, HPC experience.The estimated base salary range for this position is 200,000 to 300,000 USD per year ...

refine existing products.Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.Experience using large-scale distributed training and inference.Strong mentorship, communication, and problem-solving skills.A demonstrated passion for applied ML models and products.Bonus ...

refine existing products. Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR. Experience using large-scale distributed training and inference. Strong mentorship, communication, and problem-solving skills. A demonstrated passion for applied ...

compiler, workflow engine, distributed training system, database, debugger, or orchestration platform Meaningful contributions to systems such as Ray, Temporal, Kafka, Erlang/BEAM, PyTorch, JAX, TensorFlow, or comparable projects Strong API design judgement, open-source collaboration, and disciplined approaches to testing, versioning, compatibility, and reproducibility Experience in ML systems ...

ability to collaborate effectively in interdisciplinary teamsNice-to-haves:Experience in autonomous driving, robotics, or simulation systemsFamiliarity with large-scale training (e.g., FSDP, DeepSpeed, JAX)Experience with sim-to-real transfer or data-efficient learningContributions to open-source ML tools or research infrastructureWhat we offer you Attractive compensation with ...