Lead Engineer (HPC, GPU, CUDA)
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.
50k new jobs listed every day. Install TAL to find more jobs like this.

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
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
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
Layoffs
In the silence of the morning, a whisper cuts the air, A message cold and distant, a news too hard to bear. Dreams built on dedication, now teetering on the brink, A heart beats heavy, wondering what to think. Hands that once were busy, now tremble in the void, A future once so certain, now painfully destroyed. Colleagues turned to family, faces etched with strife, Together we once thrived, now fragments of a life. Hope flickers dim, yet stubborn in the dark, A search for new beginnings, a flame ignites a spark. For even in this sorrow, strength will find its way, In the wake of endings, dawns a brighter day.
Manhunt: CEO Murder and Social Media Backlash
- ABC to air a special on the murder of UnitedHealthcare CEO Brian Thompson and the arrest of Luigi Mangione, exploring the shocking public response. - The documentary features exclusive audio of Mangione and new investigation details, airing Dec. 19 at 10 p.m. ET on ABC and streaming on Hulu. - Mangione faces multiple charges, including first-degree murder as an act of terrorism, with potential life imprisonment without parole. Source: [New York Post](https://decider.com/2024/12/18/luigi-mangione-documentary-how-to-watch/), [The Week](https://theweek.com/politics/luigi-mangione-terrorism-charged)