Senior ML Systems Architect - TPU & AI Infra
Socket
Google's Core ML and AI Infrastructure teams advance large-scale ML across Google Cloud TPUs, enabling massive training and inference with PyTorch and JAX. We seek engineers to shape ML frameworks, compilers, and hardware-aware software across Google products. Engineers collaborate across teams, own critical ML infrastructure projects, and contribute to performance improvements on TPUs, Vertex AI, and data centers, with opportunities to switch teams as the company grows. #J-18808-Ljbffr Socket
Vacancy posted 19 hours ago
Similar jobs that could be interesting for youBased on the Senior ML Systems Architect - TPU & AI Infra in Sunnyvale, CA vacancy
- ...seeking a Software Engineer to help shape AI-native products that blend cloud, ML, and hardware. You will define robust,... ...scalable architecture and guide teams across infra, ML, and product to deliver impactful, intelligent systems. The role emphasizes technical leadership...Senior
- Google is seeking a Senior Software Engineer, AI/ML, Ads Training in Mountain View to advance core training systems and TPU efficiency. You will lead initiatives to migrate Ads models to JAX, improve scalability, and drive cross-stack optimizations with robust debugging...Senior
$170.6k - $261.3k
...join us. About the team: The AV ML Infra team at GM builds end-to-end... ...to meet the unique demands of AI and ML innovation, supporting... ...reproducibility. Together, these tools and systems empower GM to tackle the... .... Position Overview: As a Senior AI/ML Full-Stack Engineer, you...SeniorFull timeLocal areaWork from homeFlexible hours$262k - $364k
...platforms.Facilitate and drive AI transformation across Accelerator... ...high-speed IO protocols and ML SOC architecture.Google's software... ...computing, large-scale system design, networking and data storage... ...for Google Cloud customers.This senior software engineering position offers...SeniorWorldwide- ...Senior Solution Architect – AI / GPU Cloud We are seeking a Senior Solution Architect... ...understand customer AI/ML/HPC workloads, scaling requirements... .... Partner with internal infra, DC ops, and engineering... ...cluster/orchestration systems, NVIDIA GPU stack (H100/H20...Senior
GMI Cloud
Mountain View, CA2 days ago - Google is shaping the future of AI/ML hardware acceleration by driving TPU technology for demanding AI/ML tasks. You will contribute design and verification... ...focusing on TPU architecture integration within AI/ML systems. As Implementation Technical Lead, you will own chip...Senior
- A leading tech firm in Sunnyvale is seeking a Senior Staff Architect to drive cutting-edge TPU technology for AI/ML applications. This role will focus on developing innovative silicon solutions while collaborating with domain experts to enhance performance and efficiency...SeniorWorldwide
$156k - $229k
...company in Sunnyvale, CA, is seeking a Mid-level Design Verification Engineer. In this role, you will shape cutting-edge TPU technology for AI/ML applications. You will be responsible for verifying complex digital designs, collaborating with design teams, and utilizing...Senior- Google Inc. in Sunnyvale, CA seeks a Senior RTL Design Engineer to shape AI/ML hardware acceleration and TPU architecture. You will drive ASIC front-end design, translate models into bit-exact RTL, and deliver high-speed blocks for Google's TPU and cloud workloads. You...SeniorWorldwide
- Google in Sunnyvale, CA is seeking a Senior Product Manager to guide the Cloud TPU platform from conception to launch, delivering a seamless experience for... ..., and analysts to shape features and drive adoption of AI infrastructure at scale. The role focuses on defining strategy...Senior
- Google Cloud's TPU Chip Architecture and Performance team analyzes ML workloads to balance performance and cost on next-generation TPU systems. You will work with model researchers, compiler developers... ...scalable estimation methods for AI infrastructure, with emphasis on...Senior
- Google Sunnyvale is seeking an experienced RTL Design Engineer to help shape the future of AI/ML hardware acceleration. You will design and verify high-speed digital blocks for TPU architecture and contribute to bit-exact silicon implementations powering Google’s AI...Senior
- Google Inc. in Sunnyvale, CA, is seeking a Senior Engineer to lead innovative teams that shape the future of AI/ML hardware acceleration. In this role, you’ll work on cutting-edge TPU technology and influence chip design methodologies across projects. Your responsibilities...SeniorFull time
- Google Cloud in Sunnyvale, CA seeks a Senior TPU Co-Design Engineer to shape AI/ML hardware acceleration and drive TPU architecture across training and inference. You will bridge research, software, and hardware teams to maximize scalability, quality, and efficiency of...Senior
- Google is seeking a software engineer to drive TPU efficiency at scale within the Ads Training Runtime team. You will lead efforts in model stability, hardware enablement, and migration of large-scale Ads models to JAX, while optimizing across stacks for performance and...Senior
- ...world moving forward. Job Description A System Architect focusing on advanced memory technology and AI Inference solution infrastructure will lead the... ...specific trade-offs Workload Analysis: Mapping AI/ML models to hardware requirements Hardware/Software...SeniorFull timeTemporary workRemote workFlexible hoursShift work
Sandisk
Milpitas, CA12 days ago $184k - $287.5k
We are now looking for a Senior Solution Network Architect, Enterprise Products! Join the... ...design, distributed systems, and datacenter architecture... ...implementations for enterprise-grade AI systems.Your role involves... ...for enterprise AI/ML systemsCraft detailed requirements...SeniorFull timeLocal areaRemote work$152k - $241.5k
...’s three computer workloads for Physical AI. These include robotics simulation, synthetic... ...!We are seeking a hands-on Solutions Architect with deep expertise in backend infrastructure... ...Infrastructure Engineering, advancing AI/ML systems from proof of concept to production on...SeniorFull time$184k - $287.5k
...Infrastructure Specialists team is hiring a Senior Solutions Architect - AI Factory Observability & Visualization... ...the smooth functioning of HPC systems and AI factories, transforming intricate... ..., distributed systems, or large AI/ML settings.Hands-on experience with the...SeniorFull timeRemote work$184k - $287.5k
We are looking for a Senior Solutions Architect to help leading Enterprise ISVs design, build, optimize, and deploy production-grade Agentic AI systems on NVIDIA’s accelerated computing platform. In... ...Science, Electrical Engineering, AI/ML, or equivalent experience.8+ years...SeniorFull time$184k - $287.5k
...extraordinary technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing.NVIDIA is searching for an AI/ML Solutions Architect focusing on Hyperscale customers and Cloud Service Providers. Your primary...SeniorFull time$152k - $241.5k
...of a curious and motivated Senior Solutions Architect to join our NVIDIA Infrastructure... ..., and verification of AI factories, focusing on running... ...understand workload behavior and system health.Build automation... ...distributed systems, or AI/ML environments.Hands-on experience...SeniorFull timeRemote work$152k - $241.5k
...brings Artificial Intelligence (AI) to some of the biggest... ...for an experienced Solutions Architect to assist customers with adoption... ...as building Machine Learning (ML), Deep Learning (DL), Data Analytics... ...including best practices for system design, architectural patterns...SeniorFull timeLocal areaRemote work$152k - $241.5k
...the unlimited potential of AI to define the next era of computing... ...world.We are looking for a Senior Solutions Architect to support our Semiconductor... ...-generation semiconductor systems.What you’ll be doing:Support... ...e.g., NVIDIA cuLitho).Apply ML/DL to semiconductor...SeniorFull time$153.2k - $234.1k
...—from breakthrough hardware and battery systems to intuitive design, intelligent software... ...autonomous driving? Join the Embodied AI team at General Motors. Our team is developing... ...across real-world scenarios. As a Senior ML Infra Engineer, you will work on the core systems...SeniorFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours$153.2k - $234.1k
...—from breakthrough hardware and battery systems to intuitive design, intelligent software... ...autonomous driving? Join the Embodied AI Infra Foundation team at General Motors, where... ...most advanced driverless vehicles.As a Senior ML Infra Engineer, you will build critical...SeniorFull timeWork at officeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours- NVIDIA in Santa Clara, CA is seeking a senior, hands-on Solutions Architect to accelerate media broadcasting workloads using NVIDIA AI platforms. You will design end-to-end media solutions... ...SDI/IP, SMPTE 2110, and containerized AI/ML workloads is expected. #J-18808-Ljbffr...Senior
- A leading technology company in Cupertino is seeking a Senior Engineering Program Manager to drive the development of their AI platform software stack. This role requires strong technical program management skills and experience in infrastructure or platform development...Senior
- Google in Sunnyvale, CA seeks a Senior Engineer to shape AI/ML hardware acceleration. You’ll lead end-to-end chip design improvements, guide RTL verification, and drive tool flows for design teams. Join a team delivering cutting-edge TPUs powering Google services at scale...Senior
- A leading technology company in California is looking for a Senior Software Engineer to develop cutting-edge AI and ML solutions. Responsibilities include writing and testing code, collaborating through design and code reviews, and contributing to documentation. Candidates...SeniorFull time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior ML Systems Architect - TPU & AI Infra. Be the first to apply!
Related searches
- system architect Sunnyvale, CA
- technical architect Sunnyvale, CA
- pega system architect Sunnyvale, CA
- senior technology project manager Sunnyvale, CA
- remote senior business analyst Sunnyvale, CA
- senior manager clinical operations Sunnyvale, CA
- senior leadership Sunnyvale, CA
- sr. process development engineer Sunnyvale, CA
- senior manager data science Sunnyvale, CA
- senior business analyst Sunnyvale, CA