Base Career helps you apply smarter for this job.
Key skills for this role
As a GPU Systems Engineer, you’ll lead design and optimization efforts across our GPU inference stack. You will architect the libraries and runtime systems that enable Stable Diffusion, multimodal transformers, and emerging video generation models to run efficiently at scale.
You’ll guide cross-functional teams, influence hardware selection, and set the technical vision for GPU optimization practices across the company.
Key Responsibilities
Architect high-performance inference runtimes, kernel dispatchers, and memory planners for large diffusion and transformer workloads.
Lead investigations into cross-GPU performance bottlenecks, communication overheads, and scheduling inefficiencies.
Drive multi-GPU parallelism strategies — model, pipeline, and tensor parallelization.
Establish company-wide GPU optimization standards, tooling, and SLIs.
Collaborate with research to design scalable implementations of novel architectures.
Mentor engineers in profiling, tuning, and low-level optimization.
Partner with hardware vendors and infra teams to maximize cluster utilization.
Required Qualifications
5+ years in high-performance computing, GPU runtime systems, or ML infrastructure.
Proven expertise in CUDA / Triton / C++, with deep understanding of GPU scheduling, occupancy, register usage, and tensor cores.
Experience building and maintaining distributed inference or training systems.
Ability to design abstractions balancing flexibility and performance.
Strong knowledge of NCCL, NVLink, PCIe, and interconnects.
Familiar with profiling automation and performance dashboards.
Excellent technical leadership and mentoring capabilities.
Preferred Qualifications
Background in compiler-aided optimization (TVM, XLA, MLIR, Triton).
Experience tuning Stable Diffusion or transformer inference pipelines.
Exposure to heterogeneous compute backends (AMD ROCm, TPU, ASICs).
Experience working with hardware–software co-design initiatives.
Open-source or research contributions in GPU optimization
" Glance collects and processes personal data such as your name, contact details, resume and other information that may contain personal data for the purpose of processing your application. Glance utilizes Greenhouse, a third-party platform. Please review Greenhouse's Privacy Policy to understand how the data collected from you is processed and managed. By clicking on 'Submit Application', you acknowledge and agree to the above privacy terms. Should you have any privacy concerns, you may contact us through the details mentioned in your application confirmation email."
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
Bengaluru, IND
Bengaluru, IND
Raleigh, USA
New York City, USA
Bengaluru, IND
San Mateo, USA
Bengaluru, IND
Bengaluru, IND
As a GPU Systems Engineer, you’ll lead design and optimization efforts across our GPU inference stack. You will architect the libraries and runtime systems that enable Stable Diffusion, multimodal transformers, and emerging video generation models to run efficiently at scale.
You’ll guide cross-functional teams, influence hardware selection, and set the technical vision for GPU optimization practices across the company.
Key Responsibilities
Architect high-performance inference runtimes, kernel dispatchers, and memory planners for large diffusion and transformer workloads.
Lead investigations into cross-GPU performance bottlenecks, communication overheads, and scheduling inefficiencies.
Drive multi-GPU parallelism strategies — model, pipeline, and tensor parallelization.
Establish company-wide GPU optimization standards, tooling, and SLIs.
Collaborate with research to design scalable implementations of novel architectures.
Mentor engineers in profiling, tuning, and low-level optimization.
Partner with hardware vendors and infra teams to maximize cluster utilization.
Required Qualifications
5+ years in high-performance computing, GPU runtime systems, or ML infrastructure.
Proven expertise in CUDA / Triton / C++, with deep understanding of GPU scheduling, occupancy, register usage, and tensor cores.
Experience building and maintaining distributed inference or training systems.
Ability to design abstractions balancing flexibility and performance.
Strong knowledge of NCCL, NVLink, PCIe, and interconnects.
Familiar with profiling automation and performance dashboards.
Excellent technical leadership and mentoring capabilities.
Preferred Qualifications
Background in compiler-aided optimization (TVM, XLA, MLIR, Triton).
Experience tuning Stable Diffusion or transformer inference pipelines.
Exposure to heterogeneous compute backends (AMD ROCm, TPU, ASICs).
Experience working with hardware–software co-design initiatives.
Open-source or research contributions in GPU optimization
" Glance collects and processes personal data such as your name, contact details, resume and other information that may contain personal data for the purpose of processing your application. Glance utilizes Greenhouse, a third-party platform. Please review Greenhouse's Privacy Policy to understand how the data collected from you is processed and managed. By clicking on 'Submit Application', you acknowledge and agree to the above privacy terms. Should you have any privacy concerns, you may contact us through the details mentioned in your application confirmation email."
Glance is an AI-powered content discovery platform by InMobi Group that delivers personalized news, entertainment, and shopping experiences directly to smartphone lock screens across India and Southeast Asia.
Visit company websiteJobs and hiring trendsMid
Apply faster on company sites with our extension.