4 of 4 Contract JAX Jobs in the UK

Senior ML Platform Engineer - Robotics

Hiring Organisation
Harnham - Data & Analytics Recruitment
Location
London, South East, England, United Kingdom
Employment Type
Contractor
Contract Rate
£500 - £750 per day
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 ...

Senior Computer Vision Engineer - Robotics

Hiring Organisation
Pro Contract Jobs Ltd
Location
London, United Kingdom
Employment Type
Contract
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 the chance ...

Cloud Engineer

Hiring Organisation
WWT EMEA UK LIMITED
Location
Manchester, North West, United Kingdom
Employment Type
Contract
Contract Rate
£77 per hour
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 with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...

HPC AI Cloud Engineer

Hiring Organisation
WWT EMEA UK LIMITED
Location
Manchester, North West, United Kingdom
Employment Type
Contract
Contract Rate
£77 per hour
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 with … Required Skills Strong experience with GCP (Compute Engine, GKE, Storage, Networking) Hands-on with NVIDIA (CUDA/NCCL), AMD (ROCm), and TPUs (XLA/JAX/TF) Solid knowledge of HPC concepts (MPI, RDMA, InfiniBand, Slurm/Kubernetes) Experience with performance benchmarks (MLPerf, HPL, NCCL, STREAM) Proficiency in Python, Bash ...