Software Engineer, ML Compiler

Requirements:

• Bachelor's degree in Computer Science, related technical/quantitative field, or equivalent practical experience
• 2 years of experience with software development in one or more programming languages, and with data structures/algorithms
• 2 years of experience with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging)
• Experience in developing software systems in modern C++

Nice to haves:

• 3 years of experience working on compiler optimization

What you'll be doing:

• Profile and analyze the performance of compiled ML models
• Improve code generation in the XLA GPU compiler or the Triton compiler
• Serve on rotations that triage bugs, guide Public Relations from external contributors, or integrate Multi-Level Intermediate Representation (MLIR) into Google

Perks and benefits:

• No specific information provided