Senior AI/ML Training Engineer – Gen AI on TPUs/GPUs

Google LLC; Kirkland (WA); Presencial; USD 174.000 - 252.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Health insurance

Retirement benefits

Paid time off

Sick time

Maternity leave

Baby bonding leave

Holidays

Descripción de la vacante

Google LLC is seeking software engineers to advance next-gen AI workloads, including Gen AI, at scale on GCP. You will work on high-performance JAX implementations for LLMs and diffusion models, optimizing workloads across TPUs, GPUs, and CPUs in collaboration with DeepMind and Core ML teams. Individual pay includes equity and benefits.

The role supports diverse ML infrastructure, model deployment, evaluation, and data processing, with opportunities to switch projects as Google grows.

Formación

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages.
• 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
• 3 years of experience with end-to-end Machine Learning (e.g., model deployment, model evaluation, data processing, debugging).
• 3 years of generative AI or performance analysis experience.

Responsabilidades

• Write and test product or system development code.
• Collaborate with peers and stakeholders through design and code reviews to ensure best practices amongst available technologies (e.g., style guidelines, checking code in, accuracy, testability, and efficiency).
• Contribute to existing documentation or educational content and adapt content based on product/program updates and user feedback.
• Triage product or system issues and debug/track/resolve by analyzing sources of issues and the impact on hardware, network, or service operations and quality.
• Design and implement solutions in one or more specialized ML areas, leverage ML infrastructure, and demonstrate expertise in a chosen field.

Conocimientos

Software development

Testing & deployment

ML engineering

Generative AI

Performance analysis

Educación

Bachelor's degree or equivalent practical experience

Master's degree or PhD preferred

Herramientas

Docker

JAX

TPUs/GPUs

Git

Descripción del empleo

Google LLC is seeking software engineers to advance next-gen AI workloads, including Gen AI, at scale on GCP. You will work on high-performance JAX implementations for LLMs and diffusion models, optimizing workloads across TPUs, GPUs, and CPUs in collaboration with DeepMind and Core ML teams. Individual pay includes equity and benefits.

The role supports diverse ML infrastructure, model deployment, evaluation, and data processing, with opportunities to switch projects as Google grows.