High-Performance AI Kernel Engineer

4 weeks ago


Mumbai, Maharashtra, India AryaXAI Full time
About the Role

We are seeking a highly skilled High-Performance AI Kernel Engineer to join our team at AryaXAI. As a key member of our innovation team, you will be responsible for designing and developing high-performance AI kernels that power state-of-the-art AI models.

Job Description

Your primary focus will be on developing and refining low-level CUDA kernel optimizations for deep learning inference and training. You will also profile, debug, and optimize single and multi-GPU operations using tools like Nsight. Additionally, you will deeply understand and exploit GPU memory hierarchies and computational capabilities.

Key Responsibilities
  • Develop and refine low-level CUDA kernel optimizations for deep learning inference and training.
  • Profile, debug, and optimize single and multi-GPU operations using tools like Nsight.
  • Deeply understand and exploit GPU memory hierarchies and computational capabilities.
  • Implement cutting-edge methods from research papers into CUDA kernels.
Ideal Candidate Profile

We are looking for candidates with a proven track record of excellence in GPU programming and AI system optimization. The ideal candidate should have expertise in designing high-performance GeMM CUDA kernels using Tensor cores or CUDA cores, leveraging tools like CuTe or CUTLASS.

Salary Information

The estimated salary range for this position is between $120,000 and $180,000 per year, depending on experience and qualifications.



  • Mumbai, Maharashtra, India AryaXAI Full time

    At AryaXAI, we are at the forefront of innovation in AI systems that scale responsibly. As a highly skilled High-Performance AI Kernel Developer, you will play a critical role in our mission to empower researchers, engineers, and organizations with explainable, safe, and aligned AI tools.We thrive on cutting-edge innovation, collaboration, and a relentless...


  • Mumbai, Maharashtra, India AryaXAI Full time

    We are revolutionizing AI innovation at AryaXAI, a leading company in mission-critical businesses.Our mission is to create explainable, safe, and aligned AI systems that scale responsibly. We empower researchers, engineers, and organizations to unlock AI's full potential while maintaining transparency and safety.**Job Overview**We are seeking a highly...


  • Mumbai, Maharashtra, India AryaXAI Full time

    About AryaXAIAryaXAI is a pioneering force in AI innovation, dedicated to revolutionizing the field of artificial intelligence for mission-critical businesses. Our team of experts strives to build explainable, safe, and aligned systems that scale responsibly, empowering researchers, engineers, and organizations to unlock AI's full potential while maintaining...


  • Mumbai, Maharashtra, India AryaXAI Full time

    About AryaXAIAryaXAI is a leading innovator in AI, revolutionizing the field with explainable, safe, and aligned systems that scale responsibly.Job Title: High-Performance GPU Kernel Developer$170,000 - $220,000 per yearJob Description:We are seeking an exceptional GPU Kernel Developer to join our team and push the boundaries of high-performance AI...


  • Mumbai, Maharashtra, India AryaXAI Full time

    AryaXAI is at the forefront of AI innovation, pushing the boundaries of high-performance computing to create explainable, safe, and aligned systems that scale responsibly.We are a team of passionate professionals driven by a shared passion for cutting-edge innovation and collaboration. Our flat organizational structure values curiosity, initiative, and...


  • Mumbai, Maharashtra, India AryaXAI Full time

    AryaXAI is revolutionizing the field of AI by creating explainable, safe, and aligned systems that scale responsibly.We are seeking a highly skilled High-Performance AI Expert to join our team and push the boundaries of high-performance AI computation.In this role, you will design, develop, and optimize GPU kernels that power state-of-the-art AI models. Your...


  • Mumbai, Maharashtra, India AryaXAI Full time

    Company OverviewAryaXAI stands at the forefront of AI innovation, revolutionizing AI for mission-critical businesses by building explainable, safe, and aligned systems that scale responsibly.Salary Range:$175,000 - $225,000 per yearJob Description:We are seeking a highly skilled AI Researcher - GPU Kernel Developer to join our team and push the boundaries of...


  • Mumbai, Maharashtra, India AryaXAI Full time

    AryaXAI is revolutionizing AI innovation for mission-critical businesses by building explainable, safe, and aligned systems that scale responsibly. Our mission is to create AI tools that empower researchers, engineers, and organizations to unlock AI's full potential while maintaining transparency and safety.Our team thrives on a shared passion for...

  • Senior AI Engineer

    1 day ago


    Mumbai, Maharashtra, India AryaXAI Full time

    About the RoleWe are seeking a highly skilled Senior AI Engineer - GPU Optimization Specialist to join our team at AryaXAI. As a key member of our team, you will play a vital role in designing, developing, and optimizing GPU kernels that power state-of-the-art AI models.**Job Overview**Develop and refine low-level CUDA kernel optimizations for deep learning...


  • Mumbai, Maharashtra, India Tata Consultancy Services Full time

    Job OverviewWe are seeking an experienced Linux Systems Engineer to join our team at Tata Consultancy Services. As a key member of our infrastructure team, you will be responsible for designing, implementing, and maintaining high-performance Linux systems.About the RoleThe successful candidate will have a minimum of 5-8 years of hands-on experience with...


  • Mumbai, Maharashtra, India FairCloud AI Full time

    Company Overview:FairCloud AI is a pioneering startup revolutionizing the cloud infrastructure sector with innovative and efficient AI solutions that empower clients to achieve significant cost savings.Job Description:We are seeking an exceptional and talented Software Developer for an immediate start in Mumbai, India. As a developer at FairCloud AI, you...


  • Mumbai, Maharashtra, India SHI | Cymune - An SHI Company Full time

    Job Title: High-Performance Computing Systems EngineerLocation: MUMBAI, IndiaAbout the Role:We are currently seeking a skilled Technology Associate with hands-on experience in HPC and Linux systems to join our team at SHI | Cymune - An SHI Company.Responsibilities:Install, configure, and troubleshoot CentOS/RHEL/SuSE/Ubuntu operating systemsImplement and...


  • Mumbai, Maharashtra, India Wenger & Watson Inc. Full time

    We are seeking a highly skilled Senior AI Compiler Engineer to join our team at Wenger & Watson Inc.About the RoleThis is a critical position that requires expertise in developing and supporting new AI/ML compiler features and technologies to accelerate deep learning models. You will work closely with AI hardware accelerator teams to add support for compiler...


  • Mumbai, Maharashtra, India AryaXAI Full time

    Overview of AryaXAIAryaXAI is a leading innovator in AI technology, driving the development of explainable, safe, and aligned AI systems that scale responsibly. Our mission is to empower researchers, engineers, and organizations to unlock AI's full potential while maintaining transparency and safety.


  • Mumbai, Maharashtra, India Wings Inc Full time

    Join Our TeamWe are seeking an experienced .Net Engineer to join our team at Wings Inc. As a key member of our team, you will be responsible for designing and implementing high-performance software solutions using .NET framework.Responsibilities:Developing and maintaining large-scale software systemsCollaborating with cross-functional teams to drive project...


  • Mumbai, Maharashtra, India Arya Full time

    At Arya.ai, we are one of the pioneers in deep learning technology. Founded by Vinay and Deekshith from IIT Bombay in 2013, we initially provided open-source tools for deep learning before evolving into a top-tier AI PasS provider for Banks, Insurers, and Financial Services. Our team collaborates with over 100 prominent FSI logos across the globe.We address...


  • Mumbai, Maharashtra, India Employee Hub Full time

    Overview:We are looking for an experienced Senior AI Solutions Architect to join our team at Employee Hub. As a key member of our AI engineering team, you will be responsible for designing and implementing cutting-edge AI solutions that drive business value and transform our organization.About the Role:The ideal candidate will have a strong background in...


  • Mumbai, Maharashtra, India Avant Garde Corporate Services Full time

    At Avant Garde Corporate Services, we are seeking a skilled High-Performance OpenShift Engineer to join our team. This is an exceptional opportunity for a talented professional to take their career to the next level and contribute to the success of our organization. Job Overview:We are looking for a seasoned engineer with extensive experience in managing...


  • Mumbai, Maharashtra, India hard Full time

    e-tailize is at the forefront of integrating cutting-edge technology with online marketplaces and other leading platforms.We are committed to building scalable, AI-powered solutions that empower businesses to thrive in the digital age.About UsWe are a growing team of 15 talented professionals in India, where innovation and collaboration drive everything we...


  • Mumbai, Maharashtra, India Prisma AI Full time

    Job Summary:  Prisma AI seeks a skilled content developer to create engaging visual AI solutions. The ideal candidate will have a passion for technology and storytelling, with the ability to articulate complex ideas clearly.  About Us:Prisma AI pioneers advancements in computer vision technology, transforming industries through AI innovation. Our mission...