Research Intern - Maia-100 - AI Kernel Fusion

3 Months ago • 1 Years + • Research & Development • $78,600 PA - $154,560 PA

Job Summary

Job Description

This Research Internship at Microsoft focuses on AI kernel fusion techniques to optimize AI workloads for the Maia-100 AI accelerator. The intern will conduct cutting-edge research, develop and implement kernel fusion algorithms in C++ and Python, analyze experimental data, and present findings. The role requires experience with AI kernel fusion and a PhD in Computer Science or a related field. The internship is a 12-week program involving collaboration with mentors and other researchers, contributing to the development of innovative solutions for enhancing computational performance in AI. The intern will work on improving the performance of Microsoft's AI workloads by developing new kernel fusion algorithms and optimizing existing ones for various target platforms including GPUs.
Must have:
  • PhD in Computer Science or related field
  • 1+ years AI kernel fusion experience
  • Proficient in C++ and Python
  • Research and algorithm development
  • Data analysis and presentation skills
Perks:
  • Industry leading healthcare
  • Educational resources
  • Discounts on products and services
  • Savings and investments
  • Maternity and paternity leave
  • Generous time away
  • Giving programs
  • Networking opportunities

Job Details

Overview

Research Internships at Microsoft provide a dynamic environment for research careers with a network of world-class research labs led by globally-recognized scientists and engineers, who pursue innovation in a range of scientific and technical disciplines to help solve complex challenges in diverse fields, including computing, healthcare, economics, and the environment.

"Kernel Fusion" is among the most important techniques used to optimize AI workloads. The most successful fused kernels (like Flash Attention) are still discovered manually and not by automatic kernel fusion algorithms - we want to change that! We are seeking a highly motivated Research Intern to conduct cutting-edge research in AI kernel fusion techniques. This Research Internship offers a unique opportunity to work on an innovative project that could significantly improve the computational performance of Microsoft's AI workloads. 

Qualifications

Required Qualifications

  • Accepted or currently enrolled in a PhD program in Computer Science or a related field.
  • At least 1 year experience with AI kernel fusion techniques and algorithms.

Other Requirements

  • Research Interns are expected to be physically located in their manager’s Microsoft worksite location for the duration of their internship.
  • In addition to the qualifications below, you’ll need to submit a minimum of two reference letters for this position as well as a cover letter and any relevant work or research samples. After you submit your application, a request for letters may be sent to your list of references on your behalf. Note that reference letters cannot be requested until after you have submitted your application, and furthermore, that they might not be automatically requested for all candidates. You may wish to alert your letter writers in advance, so they will be ready to submit your letter. 

Preferred Qualifications

  • Proficient in C++ and Python.

The base pay range for this internship is USD $6,550 - $12,880 per month. There is a different range applicable to specific work locations, within the San Francisco Bay area and New York City metropolitan area, and the base pay range for this role in those locations is USD $8,480 - $13,920 per month.

 

Certain roles may be eligible for benefits and other compensation. Find additional benefits and pay information here: 

Microsoft accepts applications and processes offers for these roles on an ongoing basis.

Responsibilities

Research Interns put inquiry and theory into practice. Alongside fellow doctoral candidates and some of the world’s best researchers, Research Interns learn, collaborate, and network for life. Research Interns not only advance their own careers, but they also contribute to exciting research and development strides. During the 12-week internship, Research Interns are paired with mentors and expected to collaborate with other Research Interns and researchers, present findings, and contribute to the vibrant life of the community. Research internships are available in all areas of research, and are offered year-round, though they typically begin in the summer.

 

Additional Responsibilities

  • Conduct research on kernel fusion techniques to optimize computational performance.
  • Develop and implement kernel fusion algorithms for the Maia-100 AI accelerator, GPUs, and other targets.
  • Analyze and interpret experimental data, and present findings in written and oral formats.
Benefits/perks listed below may vary depending on the nature of your employment with Microsoft and the country where you work.
Industry leading healthcare
Educational resources
Discounts on products and services
Savings and investments
Maternity and paternity leave
Generous time away
Giving programs
Opportunities to network and connect

Similar Jobs

Kojima Productions - System Programmer

Kojima Productions

Tokyo, Japan (On-Site)
4 Months ago
ByteDance - Senior Software Development Engineer - NoSQL-DocumentDB

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
Nintendo - CONTRACT - Embedded Engineer, Display (NTD)

Nintendo

Redmond, Washington, United States (On-Site)
5 Months ago
PlayStation Global - Sr Software Engineer

PlayStation Global

Madison, Wisconsin, United States (Hybrid)
3 Months ago
NVIDIA - Senior Deep Learning Software Engineer, cuDNN

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
NVIDIA - Senior Applied LLM Engineer, AI – Chip Design

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
Google - Software Engineer, People with Disabilities

Google

Belo Horizonte, State Of Minas Gerais, Brazil (On-Site)
5 Months ago
Nielsen Holdings - Software Engineer - Java PL/SQL

Nielsen Holdings

Gurugram, Haryana, India (Hybrid)
3 Months ago
ByteDance - Software Engineer in Large Model System Graduate (Machine Learning Sys-US) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Haptic - Online Game Programmer

Haptic

Sweden (On-Site)
3 Months ago
Amber - 3D Level Designer (Project Based)

Amber

Brazil (On-Site)
9 Months ago
ByteDance - Senior AI Space Interaction Industrial Designer

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
MiQ - Software Engineer II

MiQ

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
NVIDIA - Senior Deep Learning Software Engineer, Inference

NVIDIA

Santa Clara, California, United States (Hybrid)
3 Months ago
Forgotten Empires - Senior Software Engineer - Age of Empires (Remote)

Forgotten Empires

Andalusia, Spain (Remote)
3 Months ago
ByteDance - Network Software Development Engineer, Network Platform

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Fluence - Controls Software Engineer-II(m/f/d)

Fluence

Berlin, Berlin, Germany (Hybrid)
6 Months ago
ByteDance - Machine Learning Engineer

ByteDance

Seattle, Washington, United States (On-Site)
3 Months ago
NVIDIA - Senior Technical Marketing Engineer - AI Infrastructure

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Redmond, Washington, United States

Payactiv - Marketing Copywriter

Payactiv

Milpitas, California, United States (Hybrid)
6 Months ago
Pipeworks - Level Designer I

Pipeworks

Eugene, Oregon, United States (Hybrid)
2 Months ago
Everyday Health Group - Senior Software Engineer, Backend - Lose It!

Everyday Health Group

United States (On-Site)
5 Months ago
Meetelise - Senior Manager, Strategic Initiatives - Office of the CEO

Meetelise

New York, New York, United States (On-Site)
5 Months ago
Alpha Sense - Director, Financial Services Account Manager

Alpha Sense

Remote, Oregon, United States (Remote)
2 Months ago
Snail Games - Interactive Films - Production Coordinator

Snail Games

Beverly Hills, California, United States (Hybrid)
2 Months ago
Scopely - Lead UI/UX Designer - WWE Champions

Scopely

California, United States (Remote)
2 Months ago
Axon - Senior Security Engineer

Axon

Scottsdale, Arizona, United States (Hybrid)
4 Months ago
Attentive - Senior Machine Learning Engineer

Attentive

San Francisco, California, United States (Hybrid)
6 Months ago
ByteDance - Research Scientist, Foundation Model, Speech & Audio

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

NVIDIA - PCB Layout Engineer - New College Graduate

NVIDIA

Shenzhen, Guangdong Province, China (On-Site)
3 Months ago
NVIDIA - ASIC Verification Engineer - GPU

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
Rivos - SOC Electrical Analysis Engineer - Full Time

Rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
6 Months ago
Tesla - Electrical Assembly Supervisor

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
2 Months ago
Netflix - Software Engineer L4, GenAI Platform

Netflix

United States (Remote)
1 Month ago
Assystems - Administrateur AVEVA PDMS E3D H/F

Assystems

Courbevoie, Île-de-France, France (Hybrid)
5 Months ago
NVIDIA - Senior Chip Architect

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
NVIDIA - Senior Signal and Power Integrity Engineer - Hardware

NVIDIA

Canada (Hybrid)
3 Months ago
Regent Craft - Mechanical Engineering Intern

Regent Craft

North Kingstown, Rhode Island, United States (On-Site)
6 Months ago
Riot Games - Principal Researcher

Riot Games

Los Angeles, California, United States (On-Site)
9 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Microsoft is a tech giant that develops, licenses, and supports a range of software products, services, and devices.
View All Jobs

Get notified when new jobs are added by Microsoft

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug