Machine Learning Compiler Engineer

Job Title: Machine Learning Compiler Engineer
Duration: Long Term
Location: Austin, TX

Responsibilities

• Work on a wide range of ML compilers
• Improve ML compiler optimization capabilities through benchmark analysis and profiling
• Innovate new ML compiler and optimization algorithms
• Incorporate generative AI into the compiler development workflow and into compilers themselves
• Upstream compiler algorithms to open-source compiler projects
• Author research publications and represent the company in conferences and industry forums
• For senior levels lead and manage projects while doing substantial technical work

Required

• Experience with compiler development and computer architecture
• ML experience
• A degree in the field of computer science or applied mathematics
• Experience with software engineering
• Solid intellectual ability, motivation, and a strong history of achievement
• Excellent oral and written communication skills
• Desired
• Experience with MLIR, MLIR dialects (LinAlg, Affine), PyTorch 2.0, TVM, Triton, and/or LLVM
• Experience incorporating generative AI into compiler development workflow (AI-assisted code generation, AI-assisted analysis, AI-assisted refactoring and review)
• Experience designing or extending DSLs as targets for AI-driven program synthesis
• ML applications and ML optimization experience
• ML architecture experience
• High-performance computing experience, including communication-avoiding algorithms
• Polyhedral compiler optimization experience
• Loop transformation and vectorization experience
• GPU programming, parallel programming experience
• General optimization experience

Minimum Qualifications:

• Bachelor's degree in Computer Science, Engineering, Information Systems, or related field and 8+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
• OR
• Master's degree in Computer Science, Engineering, Information Systems, or related field and 7+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.