Silicon Validation Engineer - LLM/PyTorch

About the organization :

QFocus AI Pvt. Ltd. (QFAI), a wholly owned subsidiary of QFocus Technologies LLC, is a consulting-led engineering services company with sharp focus on supporting next gen advanced products development across AI/ML, Compute, Communication, Storage and Consumer Electronics. Our mission is to help our customers deliver cutting-edge products on time ensuring world-class quality.

Why Join QFAI ?

Join a passionate team, dedicated to making a difference :

We are a close-knit team, with strong mission, vision and values that guide our day-to-day. Recognition of work, respect, and our multicultural community are key aspects of the employee experience and contribute to our continued success. Would you like to be part of our story? Don't hesitate, come and join us!

We are seeking an experienced Silicon Validation Engineer to join our team in validating PyTorch-based Large Language Models (LLMs) using CUDA SDK APIs. The successful candidate will be responsible for debugging, extending, and replacing the underlying CUDA code to ensure seamless functionality on our company-specific AI processors.

Key Responsibilities :

- Validate PyTorch-based LLMs on company-specific AI processors using CUDA SDK APIs

- Debug and troubleshoot issues related to CUDA code integration with PyTorch models

- Extend and modify CUDA code to optimize performance on company-specific AI processors

- Replace existing CUDA code with custom implementations to meet specific requirements

- Collaborate with cross-functional teams to ensure successful integration of LLMs with company-specific AI processors

- Develop and maintain validation frameworks and tools for PyTorch-based LLMs

- Analyze and optimize the performance of LLMs on company-specific AI processors

Requirements :

- Bachelor's or Master's degree in Computer Science, Electrical Engineering, or related fields

- Strong experience with CUDA programming and PyTorch framework

- In-depth knowledge of deep learning models, particularly Large Language Models (LLMs)

- Proficiency in C++ and Python programming languages

- Experience with debugging and troubleshooting complex software issues

- Excellent problem-solving skills and attention to detail

- Strong communication and collaboration skills

Nice to Have :

- Experience with AI processor architecture and design

- Knowledge of other deep learning frameworks, such as TensorFlow