AI Infrastructure Engineer

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, model serving frameworks)
• 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 :

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 :

Preferred Skills :

Why Join AryaXAI?

Interview Process

Main Process :

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