26 to 32 of 32 Google JAX Jobs

About Graphcore At Graphcore, we’re building the future of AI compute.We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software ...

Silicon Valley office | Reports to Founders Introduction: Join us at Fastino as we build the next generation of LLMs. Our team, boasting alumni from Google Research, Apple, Stanford, and Cambridge is on a mission to develop specialized, efficient AI. Fastino's GLiNER family of open source models has been … comparable generative AI architectures Optional - Comprehensive technical proficiency and practical experience with leading deep learning frameworks, including advanced competency in one of PyTorch, JAX, TensorFlow, or equivalent platforms for model development and optimization ...

intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join … machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down ...

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

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

frontier AI company backed by NVIDIA, building large-scale open-weight foundation models alongside researchers and engineers from DeepMind, OpenAI, Meta, Anthropic, and Google Brain. ⚡ What You’ll Do Optimise GPU performance and training efficiency across 1,000+ GPU clusters Improve utilisation, throughput, and reliability across distributed training infrastructure … accelerate large-scale model training 🔧 What They’re Looking For Deep GPU infrastructure/distributed systems experience Strong knowledge of CUDA, NCCL, PyTorch, DeepSpeed, JAX, Megatron-LM, vLLM, etc. Experience operating large-scale GPU clusters (1,000+ GPUs) Kubernetes, Slurm, or similar orchestration expertise BONUS: Experience working on NVIDIA ...

Bjak is seeking a Technical Lead in Machine Learning to oversee the execution layer of their AI chat app. This role involves managing ML systems, ensuring they are trainable and deployable under real-world constraints. ...