…

Specialist - Cloud Engineering

LTIMindtree · IT Services & Consulting

  • Bengaluru, India
  • On-site
  • Posted today
  • IT & Infrastructure

About the job

Role Description

We are seeking expertlevel GPU Software Engineers to support a highvisibility platform focused on building software tooling on top of a custom compiler and SDK

The role involves developing optimizing and porting GPU kernels and AI workloads to a specialized hardware platform

Key Responsibilities

Develop highperformance GPU kernels for specialized hardware platforms using PyTorch and Triton frameworks

Build software solutions leveraging custom compiler toolchains and SDK capabilities

Design and implement kernellevel optimizations to control and improve hardware execution behavior

Port and optimize opensource AIML models for proprietary or custom SDK environments

Adapt and run highperformance computing benchmarks and stress workloads including

High Performance Linpack HPL

BERT and similar benchmarkstyle workloads

Develop stress testing and validation workloads aligned with hardware behavior and platform reliability goals

Support validation and performance testing for both current and nextgeneration hardware platforms

Collaborate closely with platform architects compiler teams and system engineers to enhance endtoend system performance

Core Technical Skills MustHave

Programming Frameworks

Strong proficiency in Python

Expertise in CC for systemslevel programming

Handson experience with PyTorch

Experience with Triton kernel development and Triton language

GPU Systems Expertise

Mandatory experience in GPU kernel development

Strong understanding of GPU architecture and performance optimization techniques

Experience with compiler optimizations and runtime execution models

Familiarity with custom SDKs and hardware abstraction layers

Performance Workloads

Handson experience in

GEMM matrix multiplication kernel development

Porting and optimizing ML models on newcustom hardware platforms

Systemlevel performance tuning benchmarking and stress testing