Skip to main content

Customer Engineer, AI Infrastructure Modernization TPU, Google Cloud

Google
Sydney, NSW | Docklands, VIC
Full Time / Permanent

At Google, we have a vision of empowerment and equitable opportunity for all Aboriginal and Torres Strait Islander peoples and commit to building reconciliation through Google's technology, platforms and people and we welcome Indigenous applicants. Please see our Reconciliation Action Plan for more information.

Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Sydney NSW, Australia; Docklands VIC, Australia.

Minimum qualifications:

  • Bachelor's degree in computer science, mathematics, a related technical field, or equivalent practical experience.
  • 10 years of experience with cloud native architectures and modern cloud infrastructure with networking (e.g., switching/routing for ethernet/RoCE/infiniband) in customer-facing or support roles.
  • Experience developing and deploying models using deep learning frameworks (e.g., TensorFlow, PyTorch, or JAX).

Preferred qualifications:

  • Master's degree in computer science, mathematics, or a related technical field.
  • Experience as an IT infrastructure consultant or enterprise architect working in data center investment strategies and proposals.
  • Experience with AI Infrastructure systems, networking technologies (e.g., DPU, RoCE, InfiniBand), cooling, and accelerators, GPUs and TPUs.
  • Experience in leveraging main AI and software stacks and platforms to bring up and deploy AI compute clusters.
  • Knowledge of the AI infrastructure market, including main technology providers, differentiators and trends.
  • Ability to work and grow in fluid environments.

About the job

When leading companies choose Google Cloud, it's a huge win for spreading the power of cloud computing globally. Once educational institutions, government agencies, and other businesses sign on to use Google Cloud products, you come in to facilitate making their work more productive, mobile, and collaborative. You listen and deliver what is most helpful for the customer. You assist fellow sales Googlers by problem-solving key technical issues for our customers. You liaise with the product marketing management and engineering teams to stay on top of industry trends and devise enhancements to Google Cloud products.

In this role, you will work to understand the needs of customers and help shape the future using AI technology. You will work with Google Cloud Platform's technology, complete AI stack, and will position the same to customers in all verticals. You will support Google Cloud sales teams to pilot, and deploy Google Cloud's AI/ML accelerators (TPU/GPU) at AI innovators, large enterprises, and early stage AI startups. You will help customers innovate with solutions using Google Cloud's flexible and open AI infrastructure.

You will be working with Google customers on AI Infrastructure servers and networking infrastructure deployments. You will guide customer discussions on network topologies, compute/storage and support bring up of server/network/cluster/cooling deployments. You will serve as a technical expert on the Google Cloud AI infrastructure, guiding customers through the architecture, deployment, and optimization of cost-efficient training and inference jobs on Cloud TPUs.

Google Cloud accelerates every organization's ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google's cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.

Responsibilities

  • Become a trusted advisor to customers, helping them understand and incorporate AI accelerators into their overall cloud and IT strategy by designing training and inferencing platforms.
  • Demonstrate how Google Cloud is differentiated, highlighting the power of accelerators by working with customers on POCs, demonstrating features, optimizing model performance, profiling, and bench marking.
  • Design and implement multi-host AI training and inferencing solutions on Google Cloud TPUs, focusing on scalability and performance tuning.
  • Conduct performance profiling and optimization of customer models and data pipelines for the TPU architecture, identifying and resolving issues.
  • Advise customers on best practices for integrating their MLOps workflows with the Google Cloud AI Platform ecosystem for TPU utilization.

Apply for this job

Posted 1 month ago

MLOps Engineer

Akuna Capital
Sydney, NSW
  • Lead ML infrastructure, support full ML lifecycle from research to production
  • Experience building/deploying ML infrastructure in production environments
  • Python, C++, Databricks, Spark, CI/CD, model lifecycle management
Posted 23h ago

Senior Machine Learning Engineer - Training Platform (AU remote)

Canva
Sydney, NSW
hybrid
  • Build & scale Kubernetes-based ML training infrastructure at Canva
  • Senior level experience with distributed AI systems
  • Kubernetes, PyTorch, Ray, distributed training, cloud infrastructure
Posted 12d ago

Databricks Platform Engineer

Seven West Media
Sydney, NSW
hybrid
  • Administer and optimise Databricks Lakehouse platform
  • Hands-on experience administering Databricks required
  • Unity Catalog, Delta Lake, Spark, Python, SQL, Terraform, CI/CD
Posted 14d ago