Compiler Engineer – MLIR / PyTorch Infrastructure

Onsite Palo Alto, CA

pytorch compilersoftwarefull-time

lever

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will enhance the MLIR ecosystem by developing new dialects and improving interoperability with PyTorch. This role involves collaboration with hardware engineers and ML developers to create a unified stack for AI computing.

Details

Palo Alto, CA, work from office
Full-time position
Skills required: PyTorch

The work

Extend existing high-level dialects in MLIR
Design a new hardware-aware low-level dialect
Build conversion paths to current IRs
Collaborate with hardware engineers and ML developers