Software Engineer, AI/ML, Google Cloud

Job description

Minimum qualifications: - Bachelor’s degree or equivalent practical experience. - 2 years of experience programming in Python or C++. - 2 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. - 2 years of experience with one or more of the following: Speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field. - 2 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization , data processing, debugging). Preferred qualifications: - Experience with JAX transformations (vmap, pjit, grad) and the underlying XLA compiler stack. - Experience reading high level optimizer (HLO) code to understand exactly how Python code translates to hardware execution is highly valued. - Understanding of theoretical quantization and quantization techniques (PTQ, QAT, weight-only vs. activation) and low-precision numerics (int8, fp8, int4), and the mathematical implications of compression on model convergence. - Ability to interpret low-level performanc...