On-Device ML Quality Infrastructure Engineer, Graphics, Games & ML

What you'll do

• Design, build, and maintain build systems and CI/CD pipelines for the On-Device ML stack.
• Develop and maintain automation frameworks for functional, regression, and integration testing.
• Implement robustness and reliability initiatives, including fuzz testing for parsers, compilers, and runtimes.
• Track and investigate performance regressions to resolve issues across the ML stack.
• Develop and maintain tooling to help engineers diagnose, debug, and fix system issues.
• Identify gaps in testing coverage and proactively drive improvements in tooling.
• Collaborate with CI and infrastructure teams to scale and accelerate validation workflows.

What we're looking for

• Bachelor's degree in Computer Science or a relevant discipline.
• Experience with build systems such as CMake or Bazel and CI/CD pipelines.
• Strong programming and software design skills in Python.
• In-depth knowledge of quality practices, including test planning, automation, and performance evaluation.
• Excellent collaboration and communication skills.
• Master's degree in Computer Science or a relevant discipline (preferred).
• Experience with ML authoring frameworks like PyTorch, TensorFlow, or JAX (preferred).
• Experience with fuzz testing, performance profiling, or GenAI tooling (preferred).