2 of 2 Remote/Hybrid JAX Jobs in Manchester

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

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