Senior ML Compiler Engineer

femtoAI is a technology company seeking a Senior ML Compiler Engineer to design and build core components of their ML compiler. The role involves owning critical parts of the transformation and optimization pipeline, ensuring models meet hardware constraints and achieve real-time performance.

Responsibilities:

• Design and implement compiler architecture across:
• IR design and transformation pipelines
• Optimization strategies for scheduling, tiling, and memory reuse
• Lead development of complex compiler passes such as:
• Global memory allocation and liveness-driven reuse
• Cross-operator fusion and graph partitioning
• Hardware-aware scheduling strategies
• Develop cost models and optimization heuristics
• Explore advanced techniques:
• Constraint-based optimization (e.g., ILP/MILP/CP)
• Scheduling optimization
• Drive debugging of system-level issues (correctness, performance, HW mismatches)
• Collaborate with hardware teams on co-design of abstractions and execution models