Jobs on TAL
All jobsOnsiteEngineeringai/ml infra1+ yearsc++
OnsiteLeadai/ml infra

Lead Engineer (HPC, GPU, CUDA)

AIRA MatrixThane, Maharashtra, IndiaPosted 19 May 2026

AIRA Matrix is seeking a Lead Engineer to drive the performance optimization of their deep learning models. The role involves identifying architectural improvements, benchmarking computer vision algorithms, and managing the end-to-end workflow from data training to deployment. Candidates must have strong C++, Python, and GPU computing expertise, specifically in CUDA and PyTorch. This position is based onsite in Thane, India.

Matched by TAL

50k new jobs listed every day. Install TAL to find more jobs like this.

Install TAL

Experience

1+ years

Function

Engineering

Work mode

Onsite, India

Company

Tier 2

What you will work on

AIRA Matrix is seeking a Lead Engineer to drive the performance optimization of their deep learning models. The role involves identifying architectural improvements, benchmarking computer vision algorithms, and managing the end-to-end workflow from data training to deployment. Candidates must have strong C++, Python, and GPU computing expertise, specifically in CUDA and PyTorch. This position is based onsite in Thane, India.

TAL's take

Quality 60/1005/5 clarityTier 2 company

Well-defined technical role in AI acceleration, though company brand is not globally or nationally top-tier.

Highly specific technical requirements including HPC, CUDA, and deep learning framework optimization.

Must haves

  • Bachelors or Higher in Computer Science, Electrical Engineering, or related field
  • 1+ years of relevant experience in machine learning and deep neural networks
  • Strong data structures and algorithms knowledge
  • Excellent modern C++ programming skills
  • Hands-on expertise with TensorRT, CuDNN, PyTorch
  • Hand-on expertise with GPU computing (CUDA, OpenCL or OpenACC) and HPC
  • Proficient in Python programming and bash scripting

Tools and skills

c++cmakemakeclang-toolstensorrtcudnnpytorchcudaopenclopenaccmpiopenmppythonbashwindowsubuntucentos

Nice to have: ptx-isa, avx, sse, docker, singularity, shifter, charliecloud, kubernetes, slurm, lsf, ansible, puppet, salt.

About the company

Unfamiliar company in the AI/ML space, defaulting to mid-tier.

Posts mentioning AIRA Matrix