1 day ago
Base Salary
$163k - $236k/yr
Responsibilities
- Design and build demos, courses, and technical guides for Cloud TPU using JAX, PyTorch, and open-source libraries across pre-training, post-training, and inference.
- Engage AI/ML communities, maintain a professional social media presence, identify trends, mentor developers, and grow the TPU ecosystem.
- Support product launches with reference implementations, code examples, and high-visibility content for social media and conferences.
- Act as “User Zero” by testing products and partnering with product and engineering teams to improve developer experience.
- Advocate for developer needs and provide product feedback from design discussions and early testing through post-launch improvements.
Requirements
- Bachelor's degree in Computer Science, a similar technical field, or equivalent practical experience.
- Five years of experience in a technical role or equivalent technical experience.
- Experience building or optimizing models with JAX or PyTorch on TPUs or GPUs, including distributed training and hardware acceleration.
- Practical experience across pre-training, fine-tuning, and inference is preferred.
- Open-source contributions, technical conference speaking, or professional social media engagement with the AI developer community are preferred.
- Ability to own solutions, influence without authority, act as a User Zero practitioner, and communicate complex machine learning concepts through blogs, code samples, or documentation.
Benefits
- The role is based in the US and includes benefits, equity, and a 15% bonus target in addition to base pay.
Tech Stack
Google CloudPyTorch
Categories
About Google
Google builds consumer and enterprise software and services including Search, Android, YouTube, Chrome, Maps, Gmail, and Google Cloud. Its business model centers on digital advertising and paid cloud, software, and hardware offerings (e.g., Pixel and Nest) for consumers, developers, and organizations. Founded in 1998 and headquartered in Mountain View, California, Google operates globally as a subsidiary of Alphabet Inc.