5.0 - 8.0 years

55 - 60 Lacs

bengaluru

Work from Office

Key Skills: CUDA, GPU Kernels, C++, Python, GPU Optimization, Triton, ROCm, PyTorch Extensions, Distributed Inference, Mixed Precision. Roles & Responsibilities: Develop, optimize, and maintain GPU kernels (CUDA, Triton, ROCm) for diffusion, attention, and convolution operators in generative AI models. Profile end-to-end inference pipelines (data movement, kernel scheduling, memory transfers) to identify and resolve performance bottlenecks. Apply optimization techniques such as operator fusion, tiling, caching, and mixed-precision compute to maximize GPU throughput. Collaborate with research teams to productionize experimental layers or model architectures. Build benchmarking tools and micro...

Posted 6 hours ago

AI Match Score

Apply

SDE IV - GPU Engineer Glance

5.0 - 10.0 years

3 - 6 Lacs

bengaluru, karnataka, india

On-site

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...

Posted 1 day ago

AI Match Score

Apply

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.