Job Title: Machine Learning Engineer – CUDA SDK Location: Bangalore G.P.O., Karnataka (Onsite preferred) Full-time | Contract through end of year Position Overview: We are seeking a skilled Machine Learning Engineer with deep expertise in CUDA SDK to support the enhancement, porting, and validation of PyTorch-based Large Language Models (LLMs) on custom AI hardware. This is a highly technical and hands-on role that will focus on optimizing CUDA code, integrating with PyTorch models, and ensuring high performance on proprietary AI processors. Key Responsibilities: Enhance, port, and validate PyTorch-based LLMs on proprietary AI processors using CUDA SDK APIs Extend and optimize CUDA code for seamless integration and peak performance Replace existing CUDA modules with custom implementations tailored to hardware-specific requirements Debug complex issues related to CUDA-PyTorch integration and model execution Develop validation frameworks and testing tools to ensure model correctness and system stability Collaborate with cross-functional engineering teams to ensure successful deployment and optimization Profile and tune GPU kernels for speed, memory efficiency, and scalability Required Qualifications: Bachelor’s or Master’s degree in Computer Science, Electrical Engineering , or related field Proven hands-on experience with CUDA programming and optimization Deep understanding of PyTorch and familiarity with training and inference of LLMs Strong skills in C++ and Python Experience in debugging low-level CUDA and performance bottlenecks Solid understanding of deep learning fundamentals , particularly transformer-based architectures Strong problem-solving abilities and attention to detail Excellent verbal and written communication skills Preferred Qualifications (Nice to Have): Experience with custom AI processor architectures Familiarity with additional frameworks such as TensorFlow Experience with profiling tools (Nsight, nvprof, etc.) and performance tuning for GPU workloads Personal Attributes: Passion for innovation in high-performance computing and AI/ML Strong analytical and critical thinking mindset Self-driven and results-oriented Excellent collaboration skills within multidisciplinary teams Job Type: Full-time Pay: ₹3,000,000.00 - ₹5,086,392.23 per year Schedule: Day shift Monday to Friday Application Question(s): Have you worked for US Clients? Can you work on C2C? Work Location: In person