Remote GPU Kernel Evaluation Specialist - Contract

OpenTrain AI; Northern (KY); Hybrid; USD 96,000 - 124,000; Full time

Job summary

OpenTrain AI is seeking a GPU Kernel Evaluation Expert to assess the quality, correctness, and completeness of GPU and accelerator kernel development tasks used to train and evaluate frontier AI models.

You will evaluate numerical correctness, benchmarking fairness, task scope, compilation validity, and runtime behavior across a range of kernel task types. This is a fully remote contract role for eligible candidates in the United States, with a 40-hour work week and a 20+ hour minimum.

Qualifications

• 3+ years developing, optimizing, or verifying GPU kernels and accelerators.
• Experience with at least two of CUDA, Triton, NKI, or Pallas for JAX.
• Strong understanding of numerical-correctness criteria for kernels.
• Experience with Nsight, NCU, roofline analysis, or profiling tools.
• Experience with three task types: generation, translation/lowering, migration, debugging, optimization, or operator fusion.

Responsibilities

• Review GPU/kernel task submissions for quality, correctness, and completeness.
• Evaluate numerical accuracy using absolute, relative, and ULP tolerances.
• Assess selection and suitability of reference implementations.
• Analyze performance-benchmarking fairness and profiling results.
• Ensure compilation and runtime validity across environments.
• Provide rubric-based written feedback for every evaluated task.

Skills

GPU kernels

CUDA

Triton

NKI

Pallas for JAX

Tools

Nsight

NCU

Roofline analysis

Profiling tools

Job description

OpenTrain AI is seeking a GPU Kernel Evaluation Expert to assess the quality, correctness, and completeness of GPU and accelerator kernel development tasks used to train and evaluate frontier AI models.

You will evaluate numerical correctness, benchmarking fairness, task scope, compilation validity, and runtime behavior across a range of kernel task types. This is a fully remote contract role for eligible candidates in the United States, with a 40-hour work week and a 20+ hour minimum.