AryaXAI | AI Researcher

2 weeks ago

india AryaXAI Full time

AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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 cutting-edge innovation, collaboration, and a relentless drive for excellence. At AryaXAI, everyone contributes hands-on to our mission in a flat organizational structure that values curiosity, initiative, and exceptional performance.


Requirements:


  • Core Languages: CUDA, Python
  • Frameworks: CUTLASS, pybind11 (or similar)
  • Tools: Nsight, JAX/XLA bindings
  • Focus Areas: GPU Kernel Optimization, Deep Learning Inference & Training

Role Overview


We are looking for a highly skilled AI Researcher - GPU Kernel Developer 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 work will directly influence the performance and scalability of our AI systems.


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.
  • Collaborate on designing innovative solutions to achieve peak GPU performance.



Ideal Candidate Profile


We are looking for candidates with a proven track record of excellence in GPU programming and AI system optimization. You should bring:


Core Experiences:

  • Expertise in designing high-performance GeMM CUDA kernels using Tensor cores or CUDA cores, leveraging tools like CuTe or CUTLASS.
  • Proficiency in extending or writing custom attention and deep learning kernels from scratch.
  • Confidence in writing both forward and backward kernels while managing floating-point precision errors.
  • Strong optimization skills for both memory-bound and compute-bound operations.
  • Advanced knowledge of GPU architecture, including register pressure, shared-memory usage, and GPU utilization.

Preferred Skills:

  • Familiarity with profiling tools (e.g., Nsight) to identify bottlenecks and improve performance.
  • Experience integrating custom kernels with frameworks like JAX/XLA through tools like pybind11.
  • Awareness of the latest advancements in GPU optimization techniques for AI workloads.


Why Join AryaXAI?


  • Mission-Driven Impact: Work on challenges that shape the future of responsible AI.
  • Technical Excellence: Collaborate with a team of passionate and experienced professionals.
  • Growth Opportunities: Contribute across domains and expand your expertise in GPU kernel development and AI research.
  • Flexible Work Environment: Choose between remote work or relocation support to one of our key offices.


Interview Process


  • Application Review: We review your CV and a statement of exceptional work.
  • Initial Interview (15 Minutes): A technical team member will evaluate your basic skills and fit for the role.
  • Main Process:
  • Coding Assessment: Solve programming challenges in your preferred language.
  • Systems Problem-Solving: A live, hands-on session to showcase practical expertise.
  • Project Deep-Dive: Present your most notable project to our team.
  • Team Meet & Greet: Engage with the broader AryaXAI team.

Note: Our interviews are designed to conclude within one week to streamline your onboarding process.

  • AryaXAI | AI Researcher

    2 weeks ago

    india AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • AI Researcher

    2 weeks ago

    India AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • AryaXAI | Marketing Manager | india

    5 days ago

    india AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • AryaXAI | Marketing Manager | india

    4 days ago

    india AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • Marketing Manager

    6 days ago

    India AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • Marketing Manager

    5 days ago

    India AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • Marketing Manager

    4 days ago

    India AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • Marketing Manager

    4 days ago

    India AryaXAI Full time

    AryaXAI stands at the forefront of AI innovation, revolutionizing AI 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...

  • Content Strategist and Marketing Expert

    3 days ago

    India AryaXAI Full time

    Marketing Manager Job DescriptionAryaXAI is at the forefront of AI innovation, revolutionizing AI 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...

  • Cybersecurity and AI Researcher

    5 days ago

    India Ayurak AI Full time

    Job DescriptionWe are seeking a highly motivated and technically proficient Cybersecurity and AI Researcher to join our team at Ayurak AI. In this role, you will be crucial in creating standardized assessment templates for various security standards and frameworks, including NIST, SOC 2, ISO 27001, and many others.

  • AIM FIN-AI-LYTICS | Research Analyst | india

    3 weeks ago

    india AIM FIN-AI-LYTICS Full time

    About Us: At AIM FIN-AI-LYTICS , we are driven by a singular vision: to revolutionize the world of finance and valuation through the power of artificial intelligence. Guided by our mission of "Driving Financial Success with AI Precision," we aim to merge cutting-edge AI technology with the intricacies of financial modeling, valuations, and analytics to...

  • Research Scientist Computer Vision

    3 months ago

    india Murf AI Full time

    At Murf AI, we're simplifying multimedia creation by harnessing the power of artificial intelligence. Our platform empowers users to craft high-quality voiceovers effortlessly, without the need for recording equipment. Some interesting facts about Murf AI:Customers in 100+ countries1Mn+ registered users6X growth in revenue in last 12 months120+ voices in 20+...

  • AI Research Intern

    4 weeks ago

    India SuperAGI Full time

    About us:SuperAGI is pioneering the future of Artificial General Intelligence with groundbreaking research and innovative AI products. Our mission is to transform the future of applications through intelligent, autonomous solutions that drive unparalleled efficiency and growth. We are building a world where AI and human intelligence collaborate seamlessly to...

  • AI Research Intern

    4 weeks ago

    india SuperAGI Full time

    About us:SuperAGI is pioneering the future of Artificial General Intelligence with groundbreaking research and innovative AI products. Our mission is to transform the future of applications through intelligent, autonomous solutions that drive unparalleled efficiency and growth. We are building a world where AI and human intelligence collaborate seamlessly to...

  • Ayurak AI | AI Security

    5 days ago

    india Ayurak AI Full time

    (Paid Internship) -AI Security & Compliance InternCompany Description:Ayurak AI is at the forefront of cutting-edge AI solutions, developing products that make a real difference in the world. From saving lives with AI-powered emergency response to empowering businesses with data-driven insights, we're passionate about harnessing the power of AI to solve...

  • Ayurak AI | AI Security

    4 days ago

    india Ayurak AI Full time

    (Paid Internship) -AI Security & Compliance Intern Company Description: Ayurak AI is at the forefront of cutting-edge AI solutions, developing products that make a real difference in the world. From saving lives with AI-powered emergency response to empowering businesses with data-driven insights, we're passionate about harnessing the power of AI to solve...

  • AI Security

    4 days ago

    India Ayurak AI Full time

    (Paid Internship) -AI Security & Compliance InternCompany Description:Ayurak AI is at the forefront of cutting-edge AI solutions, developing products that make a real difference in the world. From saving lives with AI-powered emergency response to empowering businesses with data-driven insights, we're passionate about harnessing the power of AI to solve...

  • AI Security

    4 days ago

    India Ayurak AI Full time

    (Paid Internship) -AI Security & Compliance Intern Company Description: Ayurak AI is at the forefront of cutting-edge AI solutions, developing products that make a real difference in the world. From saving lives with AI-powered emergency response to empowering businesses with data-driven insights, we're passionate about harnessing the power of AI to solve...

  • EigenThink AI | AI/ML Engineer | india

    3 weeks ago

    india EigenThink AI Full time

    About Us:We're a small, passionate team on a mission to empower businesses with innovative AI solutions. Our focus is on developing cutting-edge applications that leverage the power of Large Language Models (LLMs) and other AI technologies to solve real-world business challenges.We're a compact team of dedicated developers looking for our next team member to...

  • EigenThink AI | AI/ML Engineer | india

    3 weeks ago

    india EigenThink AI Full time

    About Us: We're a small, passionate team on a mission to empower businesses with innovative AI solutions. Our focus is on developing cutting-edge applications that leverage the power of Large Language Models (LLMs) and other AI technologies to solve real-world business challenges. We're a compact team of dedicated developers looking for our next team member...