26 to 41 of 41 Remote JAX Jobs

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable working ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

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

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