AI/ML CUDA Engineer

- 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