Triton Compiler & NPU Performance Engineer

microTECH Global Limited; Poland; Hybrid; PLN 180,000 - 280,000; Full time

Job summary

microTECH Global Limited is seeking a world-class compiler and performance optimization expert to lead the Triton compiler and kernel framework for NPUs. You will drive AI workloads, implement advanced Triton kernels, and optimize memory, scheduling, and end-to-end performance with PyTorch and XLA.

The role demands a strong background in AI compilation, NPU programming, and systems performance engineering, with hands-on kernel development and mentoring responsibilities. PhD is a plus.

Qualifications

• Master’s degree or above in Computer Architecture, Compiler Theory, High Performance Computing, or related field; PhD preferred.
• 5+ years of experience in NPU/GPU programming, operator optimization, or compiler development.
• Deep understanding of accelerator architectures and performance bottleneck analysis (compute units, vector lanes, memory hierarchy, caches).
• Proficiency in Triton, PTX, or LLVM IR for low-level programming and optimization.
• Familiarity with PyTorch, TensorFlow, or JAX, and their graph execution and operator scheduling mechanisms.
• Proven ability to independently develop, benchmark, and optimize complex kernels.
• Strong performance profiling and analysis skills with tools like perf, torch.profiler, etc.

Responsibilities

• Lead design and development of the Triton compiler and performance optimization framework for high-performance operator implementations on NPUs.
• Implement state-of-the-art Triton kernels (Attention, MatMul, LayerNorm, Conv, Softmax) with top-tier efficiency.
• Optimize memory access patterns and parallel scheduling, understanding cache behavior, register allocation, and SM occupancy.
• Drive end-to-end performance optimization by integrating Triton with PyTorch, XLA, and runtime stacks.
• Research and apply auto-tuning, kernel fusion, and operator scheduling to maximize scalability.
• Mentor team members in Triton kernel development and establish standard optimization processes.
• Stay current with MLIR, TVM, and related compiler tech; push performance boundaries.
• Conduct performance modeling and workload fingerprinting for large models to guide system optimization.

Skills

NPU programming

GPU programming

Triton

Operator optimization

LLVM IR

PyTorch

TensorFlow

Performance profiling

Performance modeling

Kernel development

System design

Education

Master’s degree or above in Computer Architecture

Tools

Triton

PTX

LLVM IR

Job description

microTECH Global Limited is seeking a world-class compiler and performance optimization expert to lead the Triton compiler and kernel framework for NPUs. You will drive AI workloads, implement advanced Triton kernels, and optimize memory, scheduling, and end-to-end performance with PyTorch and XLA.

The role demands a strong background in AI compilation, NPU programming, and systems performance engineering, with hands-on kernel development and mentoring responsibilities. PhD is a plus.