ML Compiler Engineer, Staff - 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.

In Short

• Join a team developing the compiler backend for AI technology.
• Work on algorithmic compiler problems and hardware resource mapping.
• Collaborate with experienced compiler developers.
• Focus on model partitioning, resource allocation, and optimization.
• Deliver production quality code in modern C++.
• Experience with modern compiler infrastructures like LLVM and MLIR.
• Engage with machine learning frameworks such as ONNX, TensorFlow, and PyTorch.
• Contribute to production compiler development.
• Thrive in a fast-paced startup culture.
• Be part of an inclusive and collaborative team.

Requirements

• Bachelor's degree in Computer Science with 7+ years of experience, MSCS preferred with 5+ years.
• Ability to deliver production quality code in modern C++.
• Experience in modern compiler infrastructures, e.g., LLVM, MLIR.
• Familiarity with machine learning frameworks like ONNX, TensorFlow, and PyTorch.
• Experience in production compiler development.
• Algorithm design ability from high-level conceptual design to implementation.
• Experience with Open Source ML projects like Torch-MLIR, ONNX-MLIR, Caffe, TVM.
• Passion for a dynamic startup culture.

Benefits

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