3 of 3 Remote/Hybrid Google JAX Jobs in England

Research Engineer, Machine Learning – Paris/London/Zurich/Warsaw

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
Computer Science (or equivalent proven track record). 4 + years working on large‐scale ML codebases. Hands‐on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s). Experience in deep learning, NLP or LLMs; bonus for CUDA or data‐pipeline ...

Distinguished AI Engineer

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
/ML incubations or complex deep‐tech cross‐functional initiatives with measurable outcomes. Expertise in Python and deep experience with deep‐learning frameworks (e.g., Jax, TensorFlow, PyTorch) and state‐of‐the‐art ML techniques. Practical experience with modern AI paradigms such as world models, foundation models, multi‐modal language ...

Cloud Engineer

Hiring Organisation
WWT EMEA UK LIMITED
Location
Manchester, North West, United Kingdom
Employment Type
Contract
Contract Rate
£77 per hour
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 ...