HPC R&D Engineer

About the Role :

• We are seeking an exceptional and highly motivated HPC System R&D Engineer to join our team.
• In this role, you will be instrumental in developing and demonstrating next-generation HPC technologies, specifically focusing on their deployment and scalability across both on-premise and cloud infrastructures.
• You will identify limitations in current CPU and GPU-based solutions for large-scale AI deployments and architect innovative distributed frameworks and system-level solutions to overcome these challenges.
• Your work will directly impact the development of KLA's future tools, enabling breakthroughs in process control.
• Critically analyze existing HPC solutions based on CPU and GPU clusters to pinpoint bottlenecks and limitations in deploying AI-based solutions at scale on both on-premise and cloud infrastructures.
• Design, develop, and implement distributed frameworks and system-level solutions that enable seamless scaling of image processing and AI workloads from single GPUs to multi-node clusters with numerous GPUs.
• Focus on the challenges and opportunities of deploying HPC workloads in cloud environments, including resource management, scalability, and cost optimization.
• Install, benchmark, and rigorously evaluate pre-release hardware (CPUs, GPUs, interconnects, etc.) to assess their suitability for next-generation KLA tools.
• This includes identifying or developing relevant workloads for early-stage evaluation and prototyping.
• Conduct in-depth performance analysis of hardware and software stacks, identifying areas for optimization and improvement.
• Build functional prototypes and demonstrations of developed technologies on on-premise testbed clusters, paving the way for their integration into future KLA tools.
• Masters or PhD in Computer Science, Electrical Engineering, or a closely related field.
• Exceptional Bachelor's degree holders with significant relevant experience and an extraordinary track record will also be considered.
• Deep understanding of operating systems (Linux internals preferred), computer networks (high-speed interconnects like InfiniBand), and high-performance computing applications.
• Strong mental model of the architecture of modern distributed systems, including a comprehensive understanding of CPUs, GPUs, and various hardware accelerators.
• Proven experience with the deployment and scaling of deep-learning frameworks such as TensorFlow and PyTorch on large-scale on-premise or cloud infrastructures.
• Strong background in modern and advanced C concepts (including parallel programming paradigms).
• Excellent scripting skills in Bash, Python, or similar languages for automation, system administration, and data analysis.
• Good verbal and written communication skills to effectively collaborate with a diverse team and present technical findings.
• Things to

Make us go Wow

• Experience with the development and training of deep learning models using frameworks such as TensorFlow and PyTorch.
• Experience with building or significantly contributing to open-source operating systems and the software stack on pre-release hardware.
• Solid understanding of container infrastructure such as Docker or Singularity, and container orchestration platforms like Kubernetes for managing HPC workloads in the cloud.
• Active participation in C standards bodies or similar technical communities.
• Deep understanding and practical experience with HPC services offered by major cloud providers (AWS, Azure, GCP).
• Demonstrated expertise in performance profiling, tuning, and optimization of HPC applications.
• In-depth knowledge of high-speed interconnect technologies like InfiniBand and their impact on distributed application performance.

Location :

• Noida, Uttar Pradesh, India (Based on the current location)Benefits :
• Opportunity to be at the forefront of HPC and AI innovation for a world-leading technology company.
• Work with a team of extraordinary engineers and researchers.
• Access to state-of-the-art on-premise and cloud HPC infrastructure.
• Make a significant impact on the future of semiconductor manufacturing )