Research Assistant

Job description / Role

Description

The Modern Compilers Lab in the Computer Science program at New York University Abu Dhabi seeks to recruit a research assistant to work on the intersection of compilers and deep learning.

Many companies, such as Google, Facebook, and Amazon, are building new specialized programming frameworks. This is because these companies need to allow their users to write simple, high-level code and run it efficiently on different hardware architectures. For example, Google has built TensorFlow, a framework for deep learning allowing users to run deep learning on multiple hardware architectures without changing the code.

Our research team at NYU Abu Dhabi is developing a new programming framework called Tiramisu. Unlike existing frameworks, Tiramisu can perform advanced code optimizations that are hard to apply otherwise. Because of this, Tiramisu can generate fast code that outperforms highly optimized code written by expert programmers and can target different hardware architectures (multicore, GPUs, FPGAs, and distributed machines).

In order to have the best performance (fastest execution) for a given Tiramisu program, many code optimizations should be applied. Optimizations include vectorization (using hardware vector instructions), parallelization (running loop iterations in parallel), enhancing data locality by fusion, and blocking (i.e., accessing arrays in a way that improves temporal and spatial data locality). A large number of optimizations exist and choosing which optimization should be used and which should not is important for performance. In some situations, some optimizations are harmful to performance while they are beneficial in other situations. Currently, there is no way to help users choose which optimizations should be used. Expert programmers usually spend a lot of time trying different optimizations manually to find the best set of optimizations.

The goal of this project is to add support for automatic code optimization in Tiramisu. In particular, we want to use machine learning and deep learning to achieve this. Currently, a basic automatic optimization module that relies on machine learning has been developed and we want to take that module to the next level. The final product of this project would be a compiler pass that allows Tiramisu to automatically choose which optimization should be used for a given unoptimized program. We want to produce a high-quality technique that can be used by the users of Tiramisu and especially by our partner companies and research labs.

For consideration, applicants need to submit a cover letter, curriculum vitae, statement of research interests, and two letters of reference, all in PDF format.

The terms of employment are very competitive and include housing and educational subsidies for children. Applications will be accepted immediately and candidates will be considered until the position is filled.

About NYU Abu Dhabi

NYU Abu Dhabi is a degree-granting research university with a fully integrated liberal arts and science undergraduate program in the arts, sciences, social sciences, humanities, and engineering. NYU Abu Dhabi, NYU New York, and NYU Shanghai form the backbone of NYU’s global network university, an interconnected network of portal campuses and academic centers across six continents that enable seamless international mobility of students and faculty in their pursuit of academic and scholarly activity. This global university represents a transformative shift in higher education, one in which the intellectual and creative endeavors of academia are shaped and examined through an international and multicultural perspective. As a major intellectual hub at the crossroads of the Arab world, NYU Abu Dhabi serves as a center for scholarly thought, advanced research, knowledge creation, and sharing through its academic, research, and creative activities.