header alt image test
Microloft Logo

Microloft

LLM Inference

Reposted Yesterday
Be an Early Applicant
Remote
Hiring Remotely in United States
143K-331K Annually
Senior level
Remote
Hiring Remotely in United States
143K-331K Annually
Senior level
Work with researchers and engineers to implement and optimize generative AI inference systems. Build tools for debugging performance, profiling, distributed systems, and improve compute efficiency across large heterogeneous GPU fleets. Deliver end-to-end solutions and processes to accelerate research and production inference.
The summary above was generated by AI
Overview

Our Inference team is responsible for building and maintaining the tools and systems that enable Microsoft AI researchers to run models easily and efficiently. Our work empowers researchers to run models in RL, synthetic data generation, evals, and more. We are joint stewards of one of the largest compute fleets in the world. 

The team is responsible for optimizing compute efficiency on our heterogeneous data centers as well as enabling cutting-edge research and production deployment. We are an applied research team that is embedded directly in Microsoft AI’s research org to work as closely as possible with researchers. We are vertically integrated, owning everything from kernels to architecture co-design to distributed systems to profiling and testing tools.

This role could be a great match for you if you:
  • Understand modern generative AI architectures and how to optimize them for inference.
  • Are familiar with the internals of open-source inference frameworks like vLLM and SGLang.
  • Value clear communication, improving team processes, and being a supportive team player.
  • Are results-oriented, have a bias toward action, and enjoy owning problems end-to-end.
  • Have or can quickly gain familiarity with modern Python and its tooling, PyTorch, Nvidia GPU kernel programming and optimization, Infiniband, and NVLink.

Microsoft's mission is to empower every person and every organization on the planet to achieve more. As employees we come together with a growth mindset, innovate to empower others, and collaborate to realize our shared goals. Each day we build on our values of respect, integrity, and accountability to create a culture of inclusion where everyone can thrive at work and beyond.
Microsoft AI
This role is part of Microsoft AI. Our team is a startup-like organization within Microsoft, dedicated to pushing the boundaries of artificial intelligence while maintaining a strong commitment to safety, responsibility, and human values.
Our mission is to build AI that amplifies human potential and empowers people around the world. We strive to deliver breakthroughs that advance science, education, productivity, and global well-being.
We’re also fortunate to partner with incredible product teams giving our models the chance to reach billions of users and create immense positive impact. If you’re a brilliant, highly-ambitious and low ego individual, you’ll fit right in—come and join us as we work on our next generation of models!

Starting January 26, 2026, MAI employees are expected to work from a designated Microsoft office at least four days a week if they live within 50 miles (U.S.) or 25 miles (non-U.S., country-specific) of that location. This expectation is subject to local law and may vary by jurisdiction.


Responsibilities
  • Work alongside researchers and engineers to implement frontier AI research ideas.
  • Introduce new systems, tools, and techniques to improve model inference performance.
  • Build tools to help debug performance bottlenecks, numeric instabilities, and distributed systems issues.
  • Build tools and establish processes to enhance the team’s collective productivity.
  • Find ways to overcome roadblocks and deliver your work to users quickly and iteratively.
  • Enjoy working in a fast-paced, design-driven product development cycle.
  • Embody our Culture and Values.    

Qualifications

Required qualifications

  • Bachelor's Degree in Computer Science or related technical field AND 6+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python
    • OR equivalent experience.

Preferred qualifications

  • Master'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++, C#, Java, JavaScript, or Python OR Bachelor's Degree in Computer Science or related technical field AND 12+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python
      • OR equivalent experience.
  • Experience with generative AI.
  • Experience with distributed computing.
  • Python and Python ecosystem (eg. uv, pybind/nanobind, FastAPI) expertise.
  • Experience with large scale production inference.
  • Experience with GPU kernel programming.
  • Experience benchmarking, profiling, and optimizing PyTorch generative AI models.
  • Experience with open source inference frameworks like vLLM and SGLang.
  • Working experience and conversant with the material in the JAX scaling book. 

Software Engineering IC5 - The typical base pay range for this role across the U.S. is USD $142,800 - $274,800 per year. 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 $188,000 - $304,200 per year.

Certain roles may be eligible for benefits and other compensation. Find additional benefits and pay information here:
https://careers.microsoft.com/us/en/us-corporate-pay

Software Engineering IC6 - The typical base pay range for this role across the U.S. is USD $165,600 - $296,400 per year. 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 $220,800 - $331,200 per year.

Certain roles may be eligible for benefits and other compensation. Find additional benefits and pay information here:
https://careers.microsoft.com/us/en/us-corporate-pay


This position will be open for a minimum of 5 days, with applications accepted on an ongoing basis until the position is filled.



Microsoft is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, citizenship, color, family or medical care leave, gender identity or expression, genetic information, immigration status, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran or military status, race, ethnicity, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable local laws, regulations and ordinances. If you need assistance with religious accommodations and/or a reasonable accommodation due to a disability during the application process, read more about requesting accommodations.

Similar Jobs at Microloft

32 Minutes Ago
Remote
United States
120K-304K Annually
Mid level
120K-304K Annually
Mid level
Automation
Develop and evaluate multimodal foundation model architectures, design datasets and pipelines, run large-scale experiments, improve training/deployment efficiency, collaborate with infrastructure and product teams, and ensure responsible, safety-aligned AI development.
Top Skills: AzureCC#C++JavaJavaScriptNumpyPandasPythonPyTorchTensorFlow
32 Minutes Ago
Remote
United States
120K-261K Annually
Senior level
120K-261K Annually
Senior level
Automation
Design, implement, and debug low-level software for Azure HPC/AI VMs, focusing on hardware/software interactions, device virtualization, and GPU workload performance. Drive architecture, collaborate with partners, own reliability as an on-call DRI, and develop operational playbooks to improve stability and performance.
Top Skills: AzureCC#C++Distributed SystemsGpuHigh Performance ComputingJavaJavaScriptMachine Learning MiddlewareOperating SystemsPythonVirtual MachinesVirtualization Technologies
32 Minutes Ago
Remote
United States
120K-304K Annually
Mid level
120K-304K Annually
Mid level
Automation
Design and implement algorithms, model architectures, data mixtures, and scaling laws for large-scale pre-training. Run and oversee flagship training experiments on distributed infrastructure, collaborate with infrastructure/data/post-training teams, and iterate using rigorous, data-driven ablations to advance foundation models.
Top Skills: CC#C++JavaJavaScriptLarge-Scale Distributed SystemsPython

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account