Customer Engineer Iv, AI Infrastructure, Google Public Sector

Google Google · Big Tech · Reston, VA +2

Customer Engineer role focused on accelerating AI initiatives for Google Public Sector clients by owning the technical relationship with ML research teams, guiding them through solution design, accelerator selection, and ramping AI workloads onto Google's AI infrastructure. The role involves advising on hardware (GPU/TPU), ML frameworks, and model building techniques, acting as a hybrid technical and business advisor.

What you'd actually do

  1. Accelerate customer time-to-value on the largest AI Infrastructure and High Performance Computing (HPC) workloads in Google Public Sector.
  2. Build a trusted advisory relationship with customer architects, engineering leadership, and research teams. Identify customer priorities, technical objections and design strategies focused on Google AI Infrastructure and HPC ecosystem to deliver business value and resolve blockers.
  3. Provide domain expertise around hardware accelerators (GPU/TPU), prevailing ML Frameworks (PyTorch, Keras, JAX), and model building techniques.
  4. Make recommendations on GPU/TPU hardware, framework selection, benchmarks, and model building required to successfully implement a complete solution.
  5. Manage the holistic research engineering relationship with customers by collaborating with specialists, product management, technical teams, and more.

Skills

Required

  • Bachelor's degree in Computer Science, Mathematics, a related technical field, or equivalent practical experience.
  • 10 years of experience with cloud native architecture in a customer-facing or support role.
  • Experience with frameworks for deep learning (e.g. PyTorch, Tensor Flow, Jax, Ray, etc.), AI accelerators (e.g. TPUs, GPUs), model architectures (e.g. encoders, decoders, transformers), and using machine learning APIs.
  • Must possess an Active Top Secret/SCI US Government Security Clearance with polygraph.
  • Ability to travel up to 20% of the time as required.

Nice to have

  • Experience with prevailing ML development frameworks (e.g., Keras, PyTorch, Tensor Flow, JAX).
  • Familiarity with the AI software development life-cycle (data processing, model building, training, evaluation, deployment).
  • Familiarity with AI related tooling (Slurm, vLLM, Ray, Vertex, K8s, etc.).
  • Ability to deliver results and work cross-functionally to position and orchestrate a solution consisting of multiple products.

What the JD emphasized

  • Must possess an Active Top Secret/SCI US Government Security Clearance with polygraph.

Other signals

  • customer-facing technical role
  • AI infrastructure
  • ML research teams
  • solution design
  • customer AI workloads