Senior Developer Relations Engineer, AI Compute Enablement

Google LLC; Sunnyvale (CA); Sur place; USD 163 000 - 236 000; Plein temps

Avantages offerts par ce poste

Equity

Health benefits

Bonus target

Résumé du poste

Google LLC is seeking a Developer Relations Engineer for TPUs and open-source software to become a technical authority for the stack powering many of the world’s best models.

You will help the community navigate the full model lifecycle—from pre-training to inference—using JAX, PyTorch, and open-source libraries. Build demos, share content, and influence product development while mentoring the community.

Qualifications

• Bachelor's degree in Computer Science, a similar technical field, or equivalent practical experience.
• 5 years of experience in a technical role (e.g., software engineering, solutions consultant).
• Experience building or optimizing models using JAX or PyTorch with TPUs/GPUs, including distributed training.

Responsabilités

• Design and build demos, courses, and technical guides to inspire developers and empower them to succeed with Cloud TPU using JAX, PyTorch, and open-source libraries.
• Engage in AI/ML technical communities and maintain a professional social media presence; mentor developers and foster community that amplifies Google's TPU ecosystem.
• Support product launches by creating reference implementations, code examples, and high-visibility content for social media or conferences (Cloud Next, Google I/O).
• Act as "User Zero" and partner with product and engineering teams to improve the developer experience.
• Provide product feedback from initial design discussions and early testing to post-launch improvements.

Connaissances

JAX

PyTorch

distributed training

hardware acceleration

model deployment

Formation

Bachelor's degree in Computer Science or equivalent

Outils

TPUs/GPUs

Description du poste

Minimum qualifications:

• Bachelor's degree in Computer Science, a similar technical field, or equivalent practical experience.


• 5 years of experience in a technical role (e.g., software engineering, solutions consultant, etc.) or equivalent technical experience.


• Experience building or optimizing models using JAX or PyTorch using TPUs or GPUs, including experience with distributed training and hardware acceleration.

Preferred qualifications:

• Practical experience across the model lifecycle, including pre-training, post-training (fine-tuning), and inference (serving).


• Experience contributing to open-source projects, speaking at technical conferences, or maintaining a professional social media presence to engage with the AI developer community.


• Ability to take ownership of solutions and exert influence without authority.


• Ability to act as a \"User Zero\" practitioner, providing feedback (friction logs, bug reports) to engineering teams to improve product developer experience.


• Excellent communication skills, ability to distill complex machine learning concepts into clear, engaging content such as technical blogs, code samples, or documentation.

About the job

Our team is dedicated to making Google Cloud the best place to build and scale AI. As a Developer Relations Engineer for TPUs and open-source software, you will become a technical authority for the stack that powers many of the world's best models. You'll help the community navigate the entire model lifecycle, from pre-training and post-training to inference, using JAX, PyTorch, and open-source libraries like MaxText, Tunix, vLLM, and others. You'll build demos and create content to inspire developers and increase awareness of Google's tools, and you'll bring feedback from the community to help improve our products.

This is a high-impact role. On one side, you're an external leader: a practitioner who builds technical demos, shares them on social media, speaks at conferences, and mentors the community. On the other, you're an internal partner: our \"user zero\" who advocates for the developer community to ensure our products work well together and meet the needs of the community. You should be a strong developer with experience with open-source models and TPUs/GPUs (or a desire to learn), who enjoys building demos and other technical content and sharing them with the community, and helping developers succeed.

Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $163000 - $236000 (USD) + 15% bonus target + equity + benefits

Learn more about benefits at Google .

Responsibilities

• Design and build demos, courses, and technical guides that inspire developers and empower them to succeed with Cloud TPU using a combination of JAX, PyTorch, and open-source libraries for pre-training, post-training, and inference.


• Engage in AI/ML technical communities and maintain a professional social media presence. Identify emerging trends, mentor developers, and foster community that amplifies Google's TPU ecosystem.


• Support product launches by creating reference implementations, code examples, and high-visibility content for social media or conferences (Cloud Next, Google I/O).


• Act as \"User Zero\" and partner with product and engineering teams to improve the developer experience.


• Be an advocate of the developer community and advocate for their needs. Provide product feedback, from initial design discussions and early testing to post-launch improvements.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form .