AI Researcher

2 weeks ago

Mumbai, 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

  1. Application Review: We review your CV and a statement of exceptional work.
  2. Initial Interview (15 Minutes): A technical team member will evaluate your basic skills and fit for the role.
  3. 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.

  • Data Scientist for AI Research and Development

    7 days ago

    Mumbai, Maharashtra, India QURE AI TECHNOLOGIES PRIVATE LIMITED Full time

    Senior AI Scientist Job OverviewWe are seeking an experienced Senior AI Scientist to join our research and development team at Qure.ai, a leading healthcare technology company.

  • Research Intern

    8 months ago

    Mumbai, India Wadhwani AI Full time

    **ABOUT US** The Wadhwani Institute for Artificial Intelligence (Wadhwani AI) is the world’s first independent non-profit research institute developing AI solutions for social good. Our mission is to develop AI solutions to overcome challenges of societal importance in domains such as health, agriculture, financial inclusion, and infrastructure. Our team...

  • MavenMagnet AI | Consumer Insights Analyst

    1 week ago

    mumbai, India MavenMagnet AI Full time

    Role DescriptionMavenMagnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research.Role OverviewConduct comprehensive  qualitative market research  for a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotiveEmploy...

  • AI Research Engineer

    2 weeks ago

    Mumbai, India Sapphire Management Consultancy Full time

    Overview:We are seeking a highly skilled and innovative Research AI Engineer to drive cutting-edge AI solutions for our organization. This role is crucial in advancing our AI capabilities and fostering a culture of innovation.The successful candidate will collaborate with cross-functional teams to develop PoCs leveraging AI to address complex business...

  • MavenMagnet AI | Consumer Insights Analyst

    1 week ago

    mumbai, India MavenMagnet AI Full time

    Role Description MavenMagnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research. Role Overview Conduct comprehensive qualitative market research  for a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotive Employ...

  • AI Research Developer

    3 weeks ago

    Mumbai, Maharashtra, India Cloud Counselage Pvt Ltd Full time

    Cloud Counselage Pvt Ltd: AI Research DeveloperSalary Range: $60,000 - $80,000 per annum (dependent on location and experience)Job DescriptionWe are seeking an ambitious AI Research Developer to join our team at Cloud Counselage Pvt Ltd. As a key member of our AI research and development team, you will play a crucial role in advancing the application of...

  • AI Research Scientist

    4 months ago

    Mumbai, India Fractal Full time

    It's fun to work in a company where people truly BELIEVE in what they are doing!JD - AI Research Scientist About the Company:Fractal is one of the most prominent providers of Artificial Intelligence to Fortune 500® companies. Fractal's vision is to power every human decision in the enterprise, and bring AI, engineering, and design to help the world's...

  • Ai researcher

    3 days ago

    Mumbai, India AryaXAI Full time

    Arya XAI 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...

  • Market Research Strategist

    1 week ago

    Mumbai, Maharashtra, India MavenMagnet AI Full time

    About UsMavenMagnet AI is a cutting-edge company that empowers leading brands and organizations to make informed business decisions through actionable insights from a unified digital landscape.We leverage thousands of digitally-sourced data points, including news, blogs, forums, communities, chat rooms, product reviews, social networks, POS, product sales &...

  • Ai researcher

    2 weeks ago

    Mumbai, India AryaXAI Full time

    Arya XAI 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...

  • Business Intelligence Research Specialist

    6 days ago

    Mumbai, Maharashtra, India MavenMagnet AI Full time

    We are seeking a highly skilled Business Intelligence Research Specialist to join our team at MavenMagnet AI. As a key member of our research department, you will be responsible for conducting comprehensive qualitative market research studies to provide actionable insights for our clients.About the RoleThe ideal candidate will have 0-2 years of experience in...

  • AI Researcher

    2 weeks ago

    Mumbai, 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

    Mumbai, 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...

  • Qualitative Researcher

    3 months ago

    Mumbai, India MavenMagnet AI Full time

    Role DescriptionMavenMagnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research.Role OverviewConduct comprehensive  qualitative market research  for a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotiveEmploy...

  • Qualitative Researcher

    3 months ago

    Mumbai, India MavenMagnet AI Full time

    Role DescriptionMavenMagnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research.Role OverviewConduct comprehensivequalitative market researchfor a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotiveEmploy...

  • Python Developer and AI Researcher

    3 weeks ago

    Mumbai, Maharashtra, India Dotflick Solutions Full time

    Role: Python Data ScientistHiring Details:WFH OptionalHiring Offices: Mumbai and JaipurJob Overview:We are actively seeking talented professionals proficient in Python to join our ambitious team dedicated to pushing the frontiers of AI technology. This opportunity is tailored for individuals who thrive on developing innovative solutions and aspire to be at...

  • Consumer insights analyst

    1 week ago

    Mumbai, India MavenMagnet AI Full time

    Role DescriptionMaven Magnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research.Role OverviewConduct comprehensive  qualitative market research  for a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotiveEmploy...

  • Lead AI Researcher

    3 weeks ago

    Mumbai, Maharashtra, India QURE AI TECHNOLOGIES PRIVATE LIMITED Full time

    About Qure.ai TechnologiesWe are a leading healthcare technology company that develops innovative AI-powered solutions to diagnose and treat various health conditions.Our team is passionate about leveraging AI to make a meaningful impact on patients' lives worldwide.

  • Qualitative Researcher

    3 months ago

    Mumbai, India MavenMagnet AI Full time

    Role Description MavenMagnet AI is seeking a passionate and detail-oriented Consumer Insights Specialist with a strong background in qualitative market research. Role Overview Conduct comprehensive qualitative market research  for a diverse range of global clients in sectors such as FMCG, media, consumer durables, healthcare, and automotive Employ...

  • Insights Strategist: Consumer Market Research Expert

    1 week ago

    Mumbai, Maharashtra, India MavenMagnet AI Full time

    MavenMagnet AI is seeking a dedicated Insights Strategist to lead comprehensive consumer market research initiatives for diverse global clients across sectors such as FMCG, media, and automotive.This role involves employing innovative qualitative techniques to design and implement research studies aligned with client objectives. The Insights Strategist will...