The Senior Software Engineer, AI/ML, ML Training role at Google focuses on developing and optimizing machine learning models and architectures. The position is part of the JAX Models and Performance team, which supports Gen AI workloads on Google Cloud Platform. Engineers in this role will collaborate with peers and stakeholders to ensure best practices and contribute to high-performance implementations. This position requires versatility and leadership qualities to tackle new challenges across the full-stack.

• 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.
• 1 year of experience with software design and architecture.
• 3 years of experience with end-to-end Machine Learning.
• 3 years of generative AI or performance analysis experience.

• Write and test product or system development code.
• Collaborate with peers and stakeholders through design and code reviews.
• Contribute to existing documentation or educational content.
• Triage product or system issues and debug/track/resolve them.
• Design and implement solutions in specialized ML areas.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.