Research Assistant - AI Compiler Optimization & HPC

New York University; Al Ruways Industrial City; On-site; AED 133,920 - 200,880; Full time

Benefits offered by this job

Housing subsidy

Job summary

New York University Abu Dhabi seeks a Research Assistant - Compiler AI to advance automatic code optimization, deep learning, compiler technology, and high‑performance computing. The role involves collaborating with faculty, researchers, and industry partners on real-world optimization challenges.

You will implement compiler passes in the Tiramisu framework, develop ML models to predict optimizations, and benchmark across multicore CPUs, GPUs, and distributed environments.

Qualifications

• Bachelor's degree in Computer Science or equivalent.
• Master's degree in CS or related discipline preferred.
• Internship, academic project, or professional experience with compilers or AI/ML.
• Strong programming foundation in complex software systems.
• Familiarity with compiler concepts and ML model development.

Responsibilities

• Support research at the intersection of compiler tech, AI, DL, and HPC.
• Extend the automatic code optimization module in the Tiramisu framework.
• Develop ML models predicting compiler optimizations for programs and hardware.
• Design representations for loops, computations, memory access, dependencies, and schedules.
• Generate training data by compiling and benchmarking across configurations.
• Evaluate optimization strategies like vectorization, parallelization, loop fusion, blocking, tiling.
• Measure execution time, speedup, memory behavior, scalability, hardware utilization.
• Compare generated schedules with manual optimizations and baselines.
• Investigate cases where transformations improve or reduce performance.
• Develop compiler passes to apply optimization sequences to unoptimized code.
• Test generated programs on multicore CPUs, GPUs, FPGAs, and distributed environments.
• Build reproducible experiments, benchmarking, and model-evaluation pipelines.
• Analyze results and contribute to technical reports and conference submissions.
• Collaborate with faculty, researchers, students, labs, and industry partners.
• Participate in meetings, discussions, demonstrations, and project planning.

Skills

Compiler development

Compiler optimization

Automatic code optimization

Machine learning

Deep learning

Artificial intelligence

Tiramisu compiler

High‑performance computing

Parallel computing

Multicore programming

GPU programming

FPGA computing

Distributed computing

Vectorization

Loop parallelization

Loop fusion

Loop blocking

Loop tiling

Data locality

Intermediate representations

Code generation

Performance benchmarking

Runtime prediction

C

Python

LLVM

TensorFlow

PyTorch

Research programming

Reproducible experimentation

Education

Bachelor's degree in Computer Science

Master's degree in Computer Science or related discipline

Tools

LLVM

Job description

New York University Abu Dhabi seeks a Research Assistant - Compiler AI to advance automatic code optimization, deep learning, compiler technology, and high‑performance computing. The role involves collaborating with faculty, researchers, and industry partners on real-world optimization challenges.

You will implement compiler passes in the Tiramisu framework, develop ML models to predict optimizations, and benchmark across multicore CPUs, GPUs, and distributed environments.