Senior Staff ML Compiler Engineer - Remote

Overview

At d-Matrix, we are focused on unleashing the potential of generative AI to power the transformation of technology. We are at the forefront of software and hardware innovation, pushing the boundaries of what is possible. Our culture is one of respect and collaboration.

We value humility and believe in direct communication. Our team is inclusive, and our differing perspectives allow for better solutions. We are seeking individuals passionate about tackling challenges and are driven by execution. Ready to come find your playground? Together, we can help shape the endless possibilities of AI.

In Short

• The d-Matrix compiler team is looking for exceptional candidates to help develop the compiler backend.
• Focus on assigning hardware resources in a spatial architecture to execute low level instructions.
• Join a team of experienced compiler developers.
• Work on model partitioning, tiling, resource allocation, and optimization.
• Deliver production quality code in modern C++.
• Experience in modern compiler infrastructures like LLVM and MLIR is required.
• Familiarity with machine learning frameworks such as ONNX, TensorFlow, and PyTorch.
• Algorithm design ability from high level conceptual design to implementation.
• Passionate about thriving in a fast-paced startup culture.
• Equal Opportunity Employment Policy commitment.

Requirements

• Bachelor's degree in Computer Science with 10+ years of relevant industry experience.
• MSCS preferred with 8+ years of relevant industry experience.
• Ability to deliver production quality code in modern C++.
• Experience in modern compiler infrastructures, e.g., LLVM, MLIR.
• Experience in machine learning frameworks and interfaces, e.g., ONNX, TensorFlow, PyTorch.
• Experience in production compiler development.

Benefits

• Inclusive work environment.
• Opportunity to work on cutting-edge technology.
• Collaborative team culture.
• Focus on personal and professional growth.
• Commitment to equal opportunity employment.