Senior Software Engineer (GPU Performance)

Microsoft develops AI software and frameworks for training and deploying advanced AI models, working with next-generation supercomputers and the Maia-100 AI accelerator.
$117,200 - $229,200
Machine Learning
Senior Software Engineer
Hybrid
4+ years of experience
AI · Enterprise SaaS

Description For Senior Software Engineer (GPU Performance)

Microsoft's AI Frameworks team is seeking a Senior Software Engineer specializing in GPU Performance to join their cutting-edge initiative in developing AI software for training and deploying advanced models. This role sits at the intersection of hardware and software optimization, working directly with next-generation supercomputers and the innovative Maia-100 AI accelerator.

The position offers a unique opportunity to work on fundamental aspects of AI infrastructure, collaborating directly with OpenAI and contributing to the Azure OpenAI service. The role demands expertise in GPU performance analysis and optimization, with a focus on enabling large-scale training and inferencing of models on novel AI hardware.

As a Senior Software Engineer, you'll be responsible for developing in C/C++, Python, and GPU languages, working with state-of-the-art hardware stacks, and implementing performance optimizations for large language models. The role requires a minimum of 4 years of technical engineering experience, with deep expertise in GPU applications and kernel optimization.

The compensation is competitive, ranging from $117,200 to $229,200 per year, with higher ranges for specific locations like the San Francisco Bay area and New York City. Microsoft offers comprehensive benefits including industry-leading healthcare, educational resources, parental leave, and investment opportunities.

This is an excellent opportunity for someone passionate about AI infrastructure who wants to work at the forefront of technology, directly impacting the development of next-generation AI systems. The role combines technical depth with collaborative opportunities, working alongside researchers and developers in a fast-paced, innovation-driven environment.

Last updated 4 days ago

Responsibilities For Senior Software Engineer (GPU Performance)

  • Software development in C/C++, Python, and GPU languages such as CUDA, ROCm, or Triton
  • Work with cutting-edge hardware stacks and fast-moving software stack to deliver best-of-class inference and optimal cost
  • Engage with key partners to implement performance analysis and optimization for state-of-the-art LLMs and other models
  • Embody Microsoft's culture and values

Requirements For Senior Software Engineer (GPU Performance)

Python
Kubernetes
  • Bachelor's Degree in Computer Science or related technical field AND 4+ years technical engineering experience
  • Experience coding in C/C++, CUDA, or ROCm
  • 3+ years practical experience working on GPU applications and optimization
  • Experience writing new GPU kernels
  • Proficiency using GPU profiling tools such as NVIDIA Visual Profiler and NVIDIA Nsight Compute
  • Experience with deep learning frameworks like PyTorch, Tensorflow, or ONNX Runtime

Benefits For Senior Software Engineer (GPU Performance)

Medical Insurance
Education Budget
Parental Leave
Mental Health Assistance
  • 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

Interested in this job?

Jobs Related To Microsoft Senior Software Engineer (GPU Performance)

Senior Research Software Engineer

Senior Research Software Engineer position at Microsoft Research AI Frontiers, focusing on AI systems development and research with competitive compensation.

Senior Software Engineer

Senior Software Engineer position at Microsoft focusing on machine learning implementation and optimization, offering hybrid work in Suzhou, China.

Senior Software Engineer - AI

Senior Software Engineer position at Microsoft Security, focusing on AI-powered cybersecurity solutions with hybrid work option and competitive compensation.

Senior Research Engineer, MSR AI for Science

Senior Research Engineer position at Microsoft Research's AI for Science team, focusing on developing infrastructure and tools for AI-driven scientific research.

Senior Software Engineer

Senior Software Engineer role at Microsoft's AI Frameworks team, developing AI software and infrastructure for various platforms from supercomputers to mobile devices.