Description

Lead the architectural design and development of high-performance GPU software components and open-source AI/ML solutions.

This role is on-site.

Responsibilities

• Own the architectural design and development of GPU software components, ensuring alignment with industry standards.
• Design, write, and deliver high-quality open-source software solutions, including drivers and APIs, to enhance GPU performance.
• Conduct research on new technologies and methodologies to improve GPU efficiency and propose innovative solutions.
• Collaborate with cross-functional teams, including hardware engineers and system architects, to integrate GPU technologies successfully.
• Establish coding standards and develop comprehensive technical documentation to ensure software maintainability and scalability.

Required Skills

• 12+ years of experience in computer science or computer engineering with a strong background in GPU domain and AI/ML solutions.
• Proficiency in C++ and Python for software development and performance analysis.
• Deep understanding of GPU programming and AI/ML frameworks.
• Experience in open-source development and technical leadership.
• Strong problem-solving abilities and ability to mentor junior engineers.
• BE/B-Tech with extensive related experience, or M-Tech/PhD in Computer Science or Computer Engineering.

Preferred Skills

• Experience with GPU kernel programming using CUDA, HIP, or OpenCL.
• Experience implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Familiarity with PyTorch, TensorFlow, JAX, and containerization tools like Singularity, Docker, or Kubernetes.

Key Skills

Education

Any Graduate