Senior Chief Engineer - GPU/Artificial Intelligence (8-15 yrs)

Job Description :

Role and Responsibilities :

- Design and develop generic AI acceleration framework for GPUs, CPUs and NPUs.

- Prune and optimize trained AI models of telco use cases.

- Optimize Neural Network libraries to best adopt to underlying compute platform.

- Implement and verify the generic AI framework and optimized models.

- Profile on various compute platforms

- Define functional and performance test cases against requirement and design choices.

- Contribute to Company Intellectual Property (patents, tech papers) from the experiments and analysis done from the project.

- Work together with internal members of vRAN, L2 protocol stack and embedded system SW for product integration.

Skills and Qualification :

1) 8+ years of experience in low level embedded SW development such as architecting SW as per CPU architectures and knowledge of Assembly level debugging.

2) 5+ years of experience in in AI/ML optimization both in terms of AI model pruning and optimizing Neural Network library.

3) 5+ years of experience in training AI/ML models, fine-tuning using Tensorflow or Pytorch and integration of trained models into relevant product SW.

4) 3+ years of experience in NVIDIA GPU platforms and AI/ML ecosystem such as CUDA programming, DOCA SW framework.

5) Expertise knowledge in programming vector processing SW on SIMD processors and GPUs.

6) Proficient knowledge in hosting LLMs on GPUs and knowledge of routing AI execution jobs across CPUs and GPUs.

7) Good understanding of the High speed serial interfaces such as Ethernet, PCIe.

8) Hands-on experience in SW integration of multi-core, multi-module multi-developer environment.

Advantageous to have :

- Hands-on AVX instructions and optimizations on the assembly level

- Hands-on GPU programming and/or SIMD programming for modem Physical layer

- Experience in telecommunication and Radio Access Network SW development

Education : Bachelors or Masters or PHD