Taro Logo

Senior Software Engineer

Microsoft is a global technology company that empowers every person and organization on the planet to achieve more.
Machine Learning
Senior Software Engineer
Hybrid
5,000+ Employees
8+ years of experience
AI · Enterprise SaaS
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Senior Software Engineer

Microsoft's AI Frameworks team is seeking a Senior Software Engineer to join their cutting-edge work on AI software development. This role sits at the intersection of AI innovation and hardware optimization, working directly with OpenAI on models hosted on Azure OpenAI service. The team handles inference performance for major Microsoft products including Office, Windows, Bing, and more, serving trillions of inferences daily.

The position offers a unique opportunity to work across multiple levels of the AI software stack, from fundamental abstractions to programming models and compilers. You'll be responsible for benchmarking and optimizing OpenAI and other LLM models for performance on GPUs and Microsoft hardware, helping achieve Azure's capex goals through efficient deployment strategies.

As part of Microsoft's mission to empower every person and organization globally, you'll work in a collaborative environment with a growth mindset. The role requires strong technical expertise in high-performance computing, GPU architecture, and deep learning frameworks. You'll be working with state-of-the-art AI technology, directly impacting the performance and efficiency of some of the world's largest AI workloads.

The position offers comprehensive benefits including industry-leading healthcare, educational resources, investment options, and generous parental leave. Microsoft maintains a strong commitment to diversity and inclusion, providing equal opportunities to all qualified applicants. The hybrid work environment allows up to 50% work from home, with 0-25% travel requirements.

This is an exceptional opportunity for a seasoned software engineer passionate about AI optimization and performance engineering to make a significant impact on the future of AI infrastructure at Microsoft.

Last updated 3 months ago

Responsibilities For Senior Software Engineer

  • Identify and drive improvements to end-to-end inference performance of OpenAI and other state of the art LLMs
  • Measure, benchmark performance on Nvidia/AMD GPU's and first party Microsoft silicon
  • Optimize and monitor performance of LLMs and build SW tooling to enable insights into performance opportunities
  • Enable fast time to market of LLMs/models and their deployments at scale
  • Design, implement, and test functions or components for our AI/DNN/LLM frameworks and tools
  • Speeding up/reducing complexity of key components/pipelines to improve performance
  • Communicate and collaborate with our partners both internal and external

Requirements For Senior Software Engineer

Python
  • Bachelor's Degree in Computer Science or related technical field and 8+ years technical engineering experience with coding in languages including, but not limited to C/C++, Python or equivalent experience
  • 4+ years' practical experience working on high performance applications and performance debug and optimization on CPU's/GPU's
  • Technical background and solid foundation in software engineering principles, computer architecture, GPU architecture, HW neural net acceleration
  • Experience in end-to-end performance analysis and optimization of state of the art LLMs, HPC applications including proficiency using GPU profiling tools
  • Experience in DNN/LLM inference and experience in one or more DL frameworks such as PyTorch, Tensorflow, or ONNX Runtime and familiarity with CUDA, ROCm, Triton
  • Cross-team collaboration skills and the desire to collaborate in a team of researchers and developers

Benefits For Senior Software Engineer

Medical Insurance
Parental Leave
Education Budget
  • 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?