Postdoc in Large Language Model inferencing

KTH Royal Institute of Technology; Stockholms kommun; On-site; SEK 468,720 - 535,680; Full time

Benefits offered by this job

Good working conditions

Attractive benefits

Creative and dynamic environment

Job summary

KTH Royal Institute of Technology is offering a post-doctoral position in an ambitious project on scalable AI inference. This role involves designing and evaluating distributed systems for machine learning, working collaboratively with faculty and doctoral students, and contributing to publications. Required qualifications include a doctoral degree, strong skills in C++, Python, and experience in machine learning. This position emphasizes innovation and is set for a contract of up to two years.

Qualifications

• A doctoral degree or equivalent foreign degree is required.
• Solid background in distributed systems, networking, programming, and operating systems.
• Very good knowledge of English in speech and writing.
• Excellent critical thinking skills.
• Motivated candidate with international publications.

Responsibilities

• Designing, implementing, and evaluating distributed systems for machine learning inference.
• Collaborating with faculty members and doctoral students.
• Contributing to scientific publications and grant applications.

Skills

Distributed systems

Networking

Programming

Operating Systems

C++

Python

Linux

Scripting language

Machine learning

GPU programming

Education

Doctoral degree or equivalent

Job description

KTH Royal Institute of Technology, School of Electrical Engineering and Computer Science

The advertised post-doctoral position is within an ambitious, 5-year Wallenberg Scholar project titled "Scalable and adaptive inferencing for democratizing AI". In this 18 Million SEK project, we want to dramatically reduce the cost and power for serving large language models such as ChatGPT. More project information and our project page with early publications.

The work will primarily entail design, implementation, and evaluation of distributed systems and networks for machine learning inference. Applying machine learning concepts, with the goal of devising agentic frameworks, will also be a part of the project.

As a member of the research lab, you will be involved in experimental systems work in collaboration with faculty members and several doctoral students at our research lab. You will contribute to scientific publications as well as to new grant applications.

The work will be in collaboration with Professor Dejan Kostic and Associate Professor Marco Chiesa.

Qualifications

Requirements

• A doctoral degree or an equivalent foreign degree. This eligibility requirement must be met no later than the time the employment decision is made.
• You should also have a solid background in distributed systems, networking, programming and operating systems, and very good knowledge of English in speech and writing as it is required in day to day work.
• It is important that you have very good skills in critical thinking and very good knowledge of C++, Python, Linux, and some scripting language.
• We are seeking a highly motivated candidate with a willingness to experiment and explore, with international publications in journals and top conferences.

Preferred qualifications

• A doctoral degree or an equivalent foreign degree, obtained within the last three years prior to the application deadline
• Teaching abilities and awareness of diversity and equal opportunity issues, with specific focus on gender equality.
• You are a good collaborator but can also work independently.
• Experience with machine learning, GPU programming, or large-scale inference systems.
• Awareness of diversity and equal opportunity issues, with specific focus on gender equality

Great emphasis will be placed on personal skills.

Join us at KTH

KTH shapes the future through education, research and innovation. As a leading international technical university, we play an active role in advancing the transition towards a sustainable society. At KTH, you have the opportunity to grow and develop in a creative and dynamic environment, with good working conditions and attractive benefits. Equality, diversity and equal opportunities are essential to quality and form an integral part of KTH’s core values as a university and public authority.

Log into KTH's recruitment system to apply for this position. You are responsible for ensuring that your application is complete according to the instructions in the ad.

The application must include:

• CV including relevant professional experience and knowledge.
• Copy of diplomas and grades from your previous university studies. Translations into English or Swedish if the original documents have not been issued in any of these languages.
• Brief account of why you want to conduct research, your academic interests and how they relate to your previous studies and future goals. Maximally two pages long.

Your complete application must be received at KTH no later than the last day of application, midnight CET/CEST (Central European Time/Central European Summer Time).

About the employment

The position offered is for, at the most, two years.

It may be the case that a position at KTH is classified as a security-sensitive role in accordance with the Protective Security Act (2018:585). If this applies to the specific position, a security clearance will be conducted for the applicant in accordance with the same law with the applicant's consent. In such cases, a prerequisite for employment is that the applicant approved following the security clearance.

We firmly decline all contact with staffing and recruitment agencies and job ad salespersons.

Disclaimer: In case of discrepancy between the Swedish original and the English translation of the job announcement, the Swedish version takes precedence.