GPU Kernel Developer - Fully Remote

• Evaluate the quality and correctness of GPU/accelerator kernel development tasks for training and evaluating AI models.
• Assess numerical correctness, performance-benchmarking fairness, and task scoping across diverse kernel task types.
• Review compilation and runtime validity, providing clear, rubric-based written feedback.
• Collaborate with AI research teams to improve model training and evaluation processes.
• Work independently and asynchronously to meet deadlines and enhance AI model performance., PS: Our team reviews applications daily. Please complete your AI interview and application steps to be considered for this opportunity.

Must-Have

• 3+ years of experience with CUDA, Triton, NKI, or Pallas (JAX).
• Strong understanding of numerical-correctness criteria for kernels.
• Experience with performance profiling and benchmarking (e.g., nsight, ncu).
• Familiarity with common compilation and runtime failure modes.
• Experience with at least three kernel task types.

Preferred

• Experience with both NVIDIA GPU and custom-accelerator ecosystems.
• Background in compiler engineering, MLIR, or intermediate-representation lowering.
• Understanding of memory-hierarchy optimization.
• Contributions to kernel libraries like cuBLAS, cuDNN, or Triton., Role Name - Python Developer ROLE_DESCRIPTION - We are seeking a highly skilled Python Developer with hands-on experience in Artificial Intelligence, Machine Learning, and Gene…
• 3 days ago, ROLE_DESCRIPTION - We are seeking a highly skilled Python Developer with hands-on experience in Artificial Intelligence, Machine Learning, and Generative AI technologies. The ide…
• 3 days ago