Talent.com
High Performance Computing Software Engineer - Supercomputing
High Performance Computing Software Engineer - SupercomputingInstitute Of Foundation Models • Sunnyvale, California, United States
No longer accepting applications
High Performance Computing Software Engineer - Supercomputing

High Performance Computing Software Engineer - Supercomputing

Institute Of Foundation Models • Sunnyvale, California, United States
30+ days ago
Job type
  • Full-time
Job description
About the Institute of Foundation Models
We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy.

As part of our team, you’ll have the opportunity to work on the core of cutting-edge foundation model training, alongside world-class researchers, data scientists, and engineers, tackling the most fundamental and impactful challenges in AI development. You will participate in the development of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills will be instrumental in establishing MBZUAI as a global hub for high-performance computing in deep learning, driving impactful discoveries that inspire the next generation of AI pioneers.



The Role

IFM is building the foundational compute infrastructure that will power tomorrow’s breakthroughs in AI and computational science. We’re looking for a High Performance Computing Software Engineer to help us design, develop, and operate the software systems that run our large-scale AI workloads.

In this role, you’ll work at the intersection of high-performance computing and machine learning. You’ll be part of a team responsible for crafting the software stack that enables training of cutting-edge ML models—spanning 1000+ GPUs—and ensuring our infrastructure is robust, performant, and developer-friendly.

Job Responsibilities

    • Design and implement high-performance, distributed software solutions for large-scale AI/ML training.
    • Optimize low-level system components including Linux kernel, GPU/accelerator kernels, and interconnects.
    • Develop and tune communication libraries such as NCCL, MPI, UCX, RCCL, and RDMA-based systems.
    • Partner with ML researchers and engineers to support frameworks like PyTorch, MegatronLM, and DeepSpeed in large-scale production environments.
    • Contribute to our scheduling, orchestration, and job management systems, including Slurm and Kubernetes.
    • Debug and resolve complex issues across the stack—from kernel to container to model.
    • Work closely with hardware vendors, upstream open-source communities, and internal teams to drive performance and reliability improvements.

Skills & Experience

    • Proven experience developing and optimizing software for large-scale ML workloads (1000+ GPUs preferred).
    • Deep understanding of Linux kernel internals and accelerator (GPU) kernel development.
    • Proficiency with distributed communication libraries (e.g., NCCL, RCCL, MPI, UCX, SHARP, Libfabric).
    • Experience with ML frameworks like PyTorch, TensorFlow, JAX, or MegatronLM.
    • Strong knowledge of HPC job scheduling and orchestration tools (e.g., Slurm, Kubernetes, Pyxis).
    • Excellent debugging and systems performance tuning skills.
    • A collaborative mindset with a focus on shared success and technical excellence.
$200,000 - $400,000 a year
Visa Sponsorship
This position is eligible for visa sponsorship.

Benefits Include
*Comprehensive medical, dental, and vision benefits
*Bonus
*401K Plan
*Generous paid time off, sick leave and holidays
*Paid Parental Leave
*Employee Assistance Program
*Life insurance and disability


Create a job alert for this search

High Performance Computing Software Engineer - Supercomputing • Sunnyvale, California, United States

Similar jobs
Supercomputing Engineer

Supercomputing Engineer

ETCHED LLC • San Jose, CA, United States
Full-time
Etched is building the world's first AI inference system purpose-built for transformers - delivering over 10x higher performance and dramatically lower cost and latency than a B200.With Etched ASIC...Show more
Last updated: 5 days ago • Promoted
Senior Cloud-Scale Software Engineer - Equity Eligible

Senior Cloud-Scale Software Engineer - Equity Eligible

Fortinet, Inc. • Sunnyvale, CA, United States
Full-time
A leading cybersecurity firm is seeking a Senior/Staff Software Engineer for their FortiCNAPP Team in Sunnyvale, California.The successful candidate will design and implement scalable and resilient...Show more
Last updated: 15 days ago • Promoted
Senior, Software Engineer- Full Stack

Senior, Software Engineer- Full Stack

Walmart • Sunnyvale, CA, United States
Full-time +1
Immigration sponsorship is not available for this role.This is a full-time, onsite role at our Sunnyvale, CA office.This position is not eligible for remote work.Walmart's top-of-funnel and discove...Show more
Last updated: 5 days ago • Promoted
Principal/Senior Software Engineer, Experimentation Platform - CoreAI

Principal/Senior Software Engineer, Experimentation Platform - CoreAI

Microsoft Corporation • Mountain View, CA, United States
Full-time
CoreAI sits at the center of Microsoft's mission to redefine how software is built and experienced, providing the foundational platforms, services, and developer experiences that power the next gen...Show more
Last updated: 5 days ago • Promoted
Senior Software Engineer - HPC & Low-Level Systems

Senior Software Engineer - HPC & Low-Level Systems

Clockwork • Palo Alto, CA, United States
Full-time
Senior Software Engineer - HPC & Low-Level Systems.A Software-Driven Revolution in AI Networking.Clockwork Systems was founded by Stanford researchers and veteran systems engineers who share a visi...Show more
Last updated: 30+ days ago • Promoted
Senior Software Engineer - Compute Infrastructure (Cloud Native)

Senior Software Engineer - Compute Infrastructure (Cloud Native)

ByteDance • San Jose, CA, United States
Full-time
About the Team The Compute Infrastructure - Orchestration & Scheduling team uses Kubernetes and Serverless technologies to build a large, reliable, and efficient compute infrastructure.This infrast...Show more
Last updated: 5 days ago • Promoted
Senior Software Engineer - High-Performance Storage

Senior Software Engineer - High-Performance Storage

Pure Storage, Inc. • Santa Clara, CA, United States
Full-time
A leading data storage company in Santa Clara, CA is looking for a skilled engineer to join their Core Engineering team.You will design innovative algorithms and implement distributed systems that ...Show more
Last updated: 5 days ago • Promoted
Full-stack Software Engineer

Full-stack Software Engineer

TechSoft • Mountain View, CA, United States
Full-time
Are you up to become Full-stack Software Engineer at TechSoft?.We are looking for a passionate.Your responsibilities include gathering requirements, defining functionalities, writing and testing co...Show more
Last updated: 14 days ago • Promoted
Sr. Staff Software Engineer, Compute Infrastructure

Sr. Staff Software Engineer, Compute Infrastructure

LinkedIn • Mountain View, CA, United States
Full-time
LinkedIn is the world's largest professional network, built to create economic opportunity for every member of the global workforce.Our products help people make powerful connections, discover exci...Show more
Last updated: 1 day ago • Promoted
Software Engineer III, Full Stack, Search

Software Engineer III, Full Stack, Search

Google • Mountain View, CA, United States
Full-time
Bachelor’s degree or equivalent practical experience.Master’s degree or PhD in Computer Science or related technical field.Experience developing accessible technologies.Proficiency in code and syst...Show more
Last updated: 14 days ago • Promoted
Senior Software Engineer — AI-Driven Cloud Cost

Senior Software Engineer — AI-Driven Cloud Cost

ServiceNow, Inc. • Santa Clara, CA, United States
Full-time
A leading cloud optimization firm in California is seeking a Software Engineer to join the Cloud Cost Management team.In this role, you will design and implement scalable solutions that empower cus...Show more
Last updated: 25 days ago • Promoted
Software Development Engineer, AWS Compute Services

Software Development Engineer, AWS Compute Services

Amazon • Sunnyvale, CA, United States
Full-time
Application deadline: Applications will be accepted on an ongoing basis.Serverless Compute (https://aws.Serverless computing offers customers an exciting new way to create applications, turn busine...Show more
Last updated: 5 days ago • Promoted
Software Engineer (28558)

Software Engineer (28558)

Supermicro • San Jose, CA, United States
Full-time
Supermicro is a Top Tier provider of advanced server, storage, and networking solutions for Data Center, Cloud Computing, Enterprise IT, Hadoop/ Big Data, Hyperscale, HPC and IoT/Embedded customers...Show more
Last updated: 8 days ago • Promoted
Software Engineer, Infrastructure Performance

Software Engineer, Infrastructure Performance

OpenReq • Cupertino, CA, United States
Full-time
Etched is building AI chips that are hard-coded for individual model architectures.Our first product (Sohu) only supports transformers, but has an order of magnitude more throughput and lower laten...Show more
Last updated: 5 days ago • Promoted
Senior Software Engineer - Cloud and Kubernetes

Senior Software Engineer - Cloud and Kubernetes

NVIDIA • Santa Clara, CA, United States
Full-time
NVIDIA is seeking an exceptional Senior Software Engineer to help design and build our next-generation cloud platforms.We’re looking for a deeply technical, hands-on engineer with a broad systems p...Show more
Last updated: 5 days ago • Promoted
Cloud-Native Software Engineer - Go & Kubernetes

Cloud-Native Software Engineer - Go & Kubernetes

Spectro Cloud • San Jose, CA, United States
Full-time
A cutting-edge software firm in San Jose is seeking a talented software engineer to join their Engineering team.In this entry-level role, you will shape the Palette platform, focusing on building G...Show more
Last updated: 30+ days ago • Promoted
Senior Software Engineer - AI, Search & Knowledge Platform - Traffic Infrastructure

Senior Software Engineer - AI, Search & Knowledge Platform - Traffic Infrastructure

Apple • Cupertino, CA, United States
Full-time
Are you an expert in large-scale networking and traffic infrastructure with a passion for building next-generation platforms for machine learning systems? We’re seeking a hands-on technical leader ...Show more
Last updated: 5 days ago • Promoted
Senior Full Stack Software Engineer

Senior Full Stack Software Engineer

Peraton • Santa Clara, CA, United States
Full-time
Senior Full Stack Software Engineer.US-CA-Santa Clara | US-VA-Reston | US-MD-Lexington Park.Peraton is hiring for a Senior Full Stack Software Engineer (Angular,.NET Core, PostgreSQL, AI-Enabled De...Show more
Last updated: 5 days ago • Promoted