Description

Key Skills: C++, Python, GPU Programming, AI/ML Frameworks, Performance Analysis, Software Development, Open Source Development, Problem Solving, Technical Leadership, ROCm

Good to Have Skills: Experience with GPU kernel programming using CUDA, HIP or OpenCL. Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI). Experience in PyTorch, TensorFlow, JAX. Experience with Singularity, Docker, and/or Kubernetes. Experience with software engineering methodologies such as Agile, Scrum, Kanban.

Roles & Responsibilities:

• Own architectural design and development of GPU software components, ensuring alignment with industry standards and best practices.
• Act as one of the subject matter experts in GPU technologies, providing guidance and mentorship to junior engineers in the team on complex technical challenges.
• Design, write, and deliver high-quality open software solutions that enhance GPU performance and capabilities including developing drivers and APIs.
• Conduct research to explore new technologies and methodologies that can improve GPU performance and efficiency while proposing innovative solutions.
• Work collaboratively with cross-functional teams, including hardware engineers, system architects, and product managers to ensure successful integration of GPU technologies.
• Develop comprehensive technical documentation and establish coding standards to ensure maintainability and scalability of software products.

Experience Required: Overall 12+ years of experience in computer science or computer engineering with strong background in GPU domain and AI/ML solutions development.

Education: BE/B-Tech with several years of related experience or M-Tech with years of related experience or PhD in Computer Science or Computer Engineering or related equivalent

Key Skills

Education

Any Graduate