Domain Expert – Python
This role is part of a project supporting leading LLM companies. The primary objective is to help these foundational LLM companies improve their Large Language Models. We support companies in enhancing their models by offering high-quality proprietary data. This data can be used as a basis for fine-tuning models or as an evaluation set to benchmark the performance. Main tasks listed: Translate CUDA/C++ code into equivalent Python implementations using PyTorch and NumPy, ensuring logical and performance parity; Analyze CUDA kernels and GPU-accelerated code for structure, efficiency, and function before translation; Evaluate LLM-generated translations of CUDA/C++ code to Python, providing technical feedback and corrections; Collaborate with prompt engineers and researchers to develop test prompts that reflect real-world CUDA/PyTorch tasks.