MLIR Jobs

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence. These robots are engineered to perform a variety of tasks in home and commercial markets. We are seeking a Staff AI Inference & Acceleration Engineer to join our Platform Software team. This role will be responsible for the on-board inference architecture for our humanoid robots, acting as the technical authority on how AI workloads are mapped, optimized, and executed across the robot's compute hardware. The primary objective is to reduce power consumption and cost while meeting strict latency and reliability demands for a real-time autonomous system.

$180k - $275k

SambaNova is seeking a Compiler Architect to join their team and drive innovation in compiler infrastructure and optimization algorithms. This critical role involves working with various layers of the compiler stack and coordinating with development teams to ensure state-of-the-art ML model performance on the SambaNova platform. Responsibilities include analyzing PyTorch and machine learning models to map operations onto the company's unique dataflow hardware.

$210k - $280k

We are seeking a Member of Technical Staff for GPU Compiler Engineering to build the machine that drives our cutting-edge GPU compiler. This role involves working on the entire pipeline, from StableHLO ingestion to backend code generation and executable binary creation. You will enhance our search-based compiler infrastructure, enabling agents and RL to find optimal code forms while guaranteeing correctness. This is an opportunity to work at the foundational level of the AI stack, directly impacting the speed and cost of large-scale training runs.

$285k - $315k

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, valuing humility and direct communication. We are seeking individuals passionate about tackling challenges and driven by execution to help shape the endless possibilities of AI.

$190k - $300k

At d-Matrix, we are at the forefront of software and hardware innovation, unleashing the potential of generative AI. We are seeking passionate individuals driven by execution to join our inclusive and collaborative team. Together, we can shape the endless possibilities of AI.

$195k - $285k

At d-Matrix, we are at the forefront of software and hardware innovation, focused on unleashing the potential of generative AI. We foster a culture of respect, collaboration, humility, and direct communication, valuing diverse perspectives to create better solutions. We are seeking passionate individuals driven by execution to join our inclusive team and help shape the endless possibilities of AI.

At d-Matrix, we are at the forefront of software and hardware innovation, focused on unleashing the potential of generative AI. We are seeking passionate individuals driven by execution to join our inclusive and collaborative team. Together, we can shape the endless possibilities of AI.

Cohere

Cohere is a leading enterprise AI company building cutting-edge foundation AI models and end-to-end products. We are seeking an engineer to provide strong support for model training pipelines, ship state-of-the-art models to production, and bridge the gap between research and production. This role offers access to significant compute resources, data, and talent, with a collaborative environment where engineers contribute to both production code and research efforts based on interest and organizational needs. We are a global company with a remote-friendly approach, valuing exceptional talent regardless of location.

CA$250k - CA$535k