Embedded AI Engineer

Title: Embedded AI Engineer

Location: Remote

Engagement: Contract

Onsite Requirement: N/A

Job Summary

- Role Overview/Job Responsibilities

About this opportunity – Embedded AI Engineer We are seeking an experienced Embedded AI 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 specif...