Baber Rehman

AI compiler engineer at Huawei Hong Kong Research Center, and a PhD in programming languages from the University of Hong Kong. I build the stacks that take tile-level kernels through MLIR onto Huawei Ascend NPUs, and I work on the type systems that make language designs hold together.

GitHub · LinkedIn · baberrehman26@gmail.com · CV

Work

AI Compiler Engineer, Huawei Hong Kong Research Center (Nov 2023–present)

I work on TileLang-Ascend, Triton-Ascend, AscendNPU-IR, and Triton-distributed-ascend: the stacks that take tile-level kernels through MLIR and lower them onto the NPU.

On TileLang-Ascend I work on the AscendNPU-IR path in the open compiler: MLIR dialects and verifiers, JIT compilation, Ascend A5 operator lowering, and the tests and CI for that pipeline. On Triton-Ascend I work on the Ascend compiler backend, including static checks on kernel scope. Triton-distributed-ascend extends the same stack to distributed kernels. Related public work is a Lean model of Triton and the Triton-Ascend core, and a Lean core for Triton-Distributed.

Earlier at the same center, I created a P4-inspired DSL for programmable networking: a high-level way to describe packet-processing logic that compiles to data-plane code.

Software Engineer (DevOps), Team Lead, CloudPlex, Lahore (Jun 2015–Aug 2018). Led a team in Lahore and a remote team in Argentina. Built a cloud platform for orchestration and deployment on AWS and Azure, and automated PfSense and Fortinet configuration.

Intern, Mentor Graphics, Lahore (Jul–Sep 2014). Debugged Sourcery CodeBench and fixed GUI bugs in Java and SWT.

Research

PhD in Computer Science, University of Hong Kong (2018–2023), supervised by Bruno C. d. S. Oliveira. The dissertation is on intersection types and union types: how to combine them, when they stay disjoint, and how to keep the resulting languages type-sound.

Before that, a BS in Computer Science from PUCIT, University of the Punjab (2011–2015).

Selected publications

Also

Artifacts

Service

Program committees and artifact evaluation, newest first.

Teaching

Teaching assistant for Functional Programming and Principles of Programming Languages at the University of Hong Kong. Visiting lecturer and teaching assistant at the University of the Punjab. Resident tutor at New College, HKU, from 2019 to 2023.

Contact

Baber Rehman
baberrehman26@gmail.com
github.com/baberrehman