Lead AI Graph Compiler Engineer
modularai
Modular is seeking a Lead AI Graph Compiler Engineer to enhance its graph compiler within the MAX Framework, which executes machine learning models efficiently across various hardware platforms. The role involves setting technical direction, collaborating with teams, and addressing complex challenges in AI infrastructure development.
Qualification
- Strong experience with compiler design and implementation, particularly using MLIR and LLVM.
- Proficiency in programming languages relevant to compiler development, such as C++ or Rust.
- Experience with machine learning frameworks and concepts.
- Ability to work collaboratively in a team environment and communicate effectively with technical and non-technical stakeholders.
- A passion for AI and a desire to innovate in the field of AI infrastructure.
Responsibility
- Set the technical direction for a team building a heterogeneous multi-device compiler and runtime system using MLIR and LLVM technologies.
- Collaborate with the Mojo compiler team to leverage Mojo's compile-time meta-programming capabilities.
- Develop expertise in new hardware platforms and work with Mojo kernel developers for optimal performance.
- Address complex technical challenges to support ML engineers and expedite model development.
- Contribute to the design and implementation of new features for the graph compiler.
