Research Intern - AI Systems and Tools

Microsoft Microsoft · Big Tech · Redmond, WA +1 · Applied Sciences

Research Intern role focused on developing AI systems and tools, particularly developer tools for Microsoft's custom Maia AI hardware. This involves working on profilers, debuggers, performance analysis tools, and simulators to enable efficient execution of AI models on AI accelerators. The role collaborates with AI researchers, hardware teams, and AI compilers teams, and involves work on device firmware, host software, and integration with AI/ML frameworks.

What you'd actually do

  1. work on multiple levels of the AI system and infrastructure stack, with an emphasis on developer tools for Microsoft's custom Maia AI hardware.
  2. build new tools that help developers author highly efficient kernels for state-of-the-art models to execute on AI accelerators.
  3. collaborate with partner teams to build new tools that help developers author highly efficient kernels for state-of-the-art models to execute on AI accelerators.
  4. work with partner teams to build new tools that help developers author highly efficient kernels for state-of-the-art models to execute on AI accelerators.
  5. collaborate with other Research Interns and researchers, present findings, and contribute to the vibrant life of the community.

Skills

Required

  • PhD program in Computer Science, Computer Engineering, Electrical Engineering or related STEM field
  • research experience
  • peer-reviewed publications
  • software development
  • C++

Nice to have

  • developer tools for accelerators (profilers, debuggers, etc.)
  • AI Compilers
  • systems and parallel programming
  • complex problems related to software development
  • effectively collaborate with your mentor and team members and partners

What the JD emphasized

  • Currently enrolled in PhD program in Computer Science, Computer Engineering, Electrical Engineering or related STEM field.
  • At least 1 year of experience in conducting research, writing peer-reviewed publications and software development.
  • At least 1 year of experience with software development in C++.

Other signals

  • AI accelerators
  • developer tools for AI hardware
  • optimize and scale out model training and inference
  • software stacks for AI accelerators
  • custom Maia AI hardware