High-Performance AI Researcher

1 week ago


Mumbai, Maharashtra, India AryaXAI Full time
AryaXAI: Revolutionizing AI with Cutting-Edge Innovation
At AryaXAI, we are at the forefront of AI innovation, pushing the boundaries of high-performance AI computation. Our mission is to create AI tools that empower researchers, engineers, and organizations to unlock AI's full potential while maintaining transparency and safety.

We thrive on a shared passion for cutting-edge innovation, collaboration, and a relentless drive for excellence. Our flat organizational structure values curiosity, initiative, and exceptional performance, allowing everyone to contribute hands-on to our mission.

The Role Overview:

We are seeking a highly skilled High-Performance AI Researcher - GPU Kernel Developer to join our team and make a significant impact in the field of AI optimization. In this role, you will design, develop, and optimize GPU kernels that power state-of-the-art AI models, directly influencing the performance and scalability of our AI systems.

Key Responsibilities:

- Develop and refine low-level CUDA kernel optimizations for deep learning inference and training, leveraging expert knowledge of GPU architecture and memory hierarchies.
- Profile, debug, and optimize single and multi-GPU operations using advanced tools like Nsight, ensuring peak performance and efficiency.
- Collaborate with our team to design innovative solutions to achieve peak GPU performance, staying up-to-date with the latest advancements in GPU optimization techniques for AI workloads.

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 bring:

- 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, managing floating-point precision errors with confidence.
- Strong optimization skills for both memory-bound and compute-bound operations, with advanced knowledge of GPU architecture and utilization.

Preferred Skills:

- Familiarity with profiling tools (e.g., Nsight) to identify bottlenecks and improve performance, ensuring seamless integration with frameworks like JAX/XLA through tools like pybind11.
- Awareness of the latest advancements in GPU optimization techniques for AI workloads, staying ahead of the curve in this rapidly evolving field.

Benefits and Opportunities:

- Competitive Salary: $180,000 per annum, reflecting the expertise and dedication required for this role.
- Mission-Driven Impact: Work on challenges that shape the future of responsible AI, contributing to the development of explainable, safe, and aligned AI systems.
- Technical Excellence: Collaborate with a team of passionate and experienced professionals, driving innovation and excellence in AI research and development.
- Growth Opportunities: Contribute across domains and expand your expertise in GPU kernel development and AI research, with flexible work environments and relocation support available.

Interview Process:

Our interview process is designed to evaluate your technical skills, fit for the role, and ability to contribute to our mission-driven team. We use a combination of coding assessments, systems problem-solving exercises, and project deep-dives to ensure you have the necessary expertise and passion for this role.

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


  • 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.


  • Mumbai, Maharashtra, India Mitra AI Full time

    About Mitra AIMitra AI is a global technology company that specializes in innovative AI-driven solutions, cloud engineering, enterprise integration, and workflow automation.We are headquartered in the UK and have been serving global clients for over a decade across industries such as BFSI, telecommunications, healthcare, and retail, helping businesses...

  • AI Research Developer

    2 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...


  • 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 &...


  • 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...


  • Mumbai, Maharashtra, India Jio Haptik Full time

    We are looking for a highly skilled DevOps Engineer to join our team at Jio Haptik. This is an excellent opportunity to work with a leading Conversational AI company that has reached over 500+ million devices and processed over 4 billion conversations.Job Description:Build, manage, and maintain large-scale systems and architecture for our platform, spanning...


  • 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...

  • Lead AI Researcher

    2 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.


  • Mumbai, Maharashtra, India AryaXAI Full time

    About UsAryaXAI 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.We thrive on a...


  • Mumbai, Maharashtra, India Skill Secure AI Full time

    About the Role:We are seeking a highly motivated AI Development Specialist to join our team at Skill Secure AI. In this role, you will work alongside experienced AI engineers and data scientists to design, develop, and implement AI models and algorithms.Key Responsibilities:Assist in the development and implementation of machine learning models and...


  • Mumbai, Maharashtra, India Arya Full time

    AryaXAI is pioneering AI innovation, transforming mission-critical businesses with explainable, safe, and aligned systems that scale responsibly. Our mission is to create AI tools empowering 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...


  • 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...

  • Senior Data Scientist

    2 weeks ago


    Mumbai, Maharashtra, India Snaphunt Full time

    We are seeking a highly skilled and experienced Senior Data Scientist - AI Researcher to join our team at Snaphunt, a leading peer-to-peer lending platform.Job DescriptionWe are looking for an exceptional candidate with expertise in deep learning and machine learning techniques to design and develop cutting-edge AI solutions. The successful candidate will...

  • AI/ML Consultant

    3 weeks ago


    Mumbai, Maharashtra, India Antal International Full time

    Job Title: AI/ML Consultant - High-End AI Strategy and ConsultingWe are seeking an experienced AI/ML Consultant to join our high-end AI strategy and consulting team. As a seasoned professional, you will work closely with clients to design, develop, and implement AI solutions that enhance organizational efficiency.


  • Mumbai, Maharashtra, India Cloud Counselage Pvt Ltd Full time

    Job OverviewWe are seeking a highly motivated and talented AI Research Assistant to join our team at Cloud Counselage Pvt Ltd. As an AI Research Assistant, you will have the opportunity to work closely with our experienced professionals in the field of artificial intelligence to support various projects aimed at advancing the application of AI in specific...


  • Mumbai, Maharashtra, India Lexlegis AI Full time

    We are seeking a seasoned Senior User Experience Designer to join our team at Lexlegis AI. As a key member of our design team, you will be responsible for crafting intuitive and engaging user interfaces that meet the needs of our users.Job DescriptionAbout Us:Lexlegis AI is a cutting-edge technology company dedicated to revolutionizing the legal industry...

  • Senior AI Researcher

    3 weeks ago


    Mumbai, Maharashtra, India L&T Financial Services Full time

    About CompanyL&T Finance is a leading non-banking financial company (NBFC) in India, part of the larger Larsen & Toubro (L&T) Group. Our company has grown into a diversified financial services provider, offering a wide range of products including retail and corporate finance, housing finance, asset management, and wealth management services.We are committed...


  • Mumbai, Maharashtra, India Third Bridge Full time

    Unlock the Power of Private Equity with Third BridgeSalary: $80,000 - $100,000 per yearIn a fast-paced and dynamic industry, our team at Third Bridge is seeking a highly skilled and ambitious Private Equity Research Associate to join our ranks. As a key member of our team, you will be responsible for providing exceptional research support to our private...


  • Mumbai, Maharashtra, India Runaway Insights Full time

    Job SummaryAs a Research Associate at Runaway Insights, you will work alongside the Primary Consultant to conduct online research, analyze data, and support report preparation. Key responsibilities include identifying relevant sources, utilizing tools like Google and AI-based platforms, and organizing findings in various formats such as Excel, PowerPoint, or...