waypointjobs

Confidential

GPU Kernel Developer for AI Model Evaluation

Fellsburg, PA

Check who can apply and the requirements below before continuing.

About this opportunity

Confidential lists this GPU Kernel Developer for AI Model Evaluation opportunity in Fellsburg, Pennsylvania. Review the employer’s description below for duties, qualifications and application requirements.

Job description

Role Overview

Assess GPU and accelerator kernel development tasks that support the training and evaluation of advanced AI models. You will review task quality, numerical correctness, completeness, benchmarking fairness, scope, and compilation and runtime validity across a range of kernel-development scenarios, then deliver clear written feedback using defined evaluation rubrics.

Key Responsibilities

Evaluate GPU and accelerator kernel tasks for quality, correctness, completeness, and appropriate scope.

Review numerical-validation approaches, performance-benchmarking fairness, and compilation and runtime validity.

Provide precise, rubric-based written feedback across diverse kernel task types.

Qualifications

3+ years of hands-on experience developing, optimizing, or verifying GPU or accelerator kernels in at least two of CUDA, Triton, NKI, or Pallas for JAX.

Strong knowledge of kernel numerical-correctness standards, including absolute, relative, and ULP tolerances and selection of reference implementations.

Experience profiling and benchmarking performance with tools such as Nsight, NCU, roofline analysis, or framework-native profilers.

Familiarity with compilation and runtime failure modes, including driver mismatches, out-of-memory errors, launch-configuration errors, shape or stride mismatches, and autotuning failures.

Experience with at least three kernel task types: specification-based generation, cross-framework translation or lowering, hardware-target migration, debugging, performance optimization, or operator fusion.

Preferred Qualifications

Experience in both NVIDIA GPU environments, including CUDA or Triton, and custom accelerator environments, including NKI, Pallas, or TPU.

Background in compiler engineering, MLIR, or intermediate-representation lowering.

Knowledge of memory-hierarchy optimization, including shared-memory tiling, register pressure, bank conflicts, and coalescing patterns.

Contributions to kernel libraries such as cuBLAS, cuDNN, Triton community kernels, or JAX/XLA custom calls.

Work Terms

Remote role open to candidates located in the United States.

Hourly engagement.

Compensation

$70 to $90 per hour.

Who can apply

Review the original listing for work authorization, qualifications and employer requirements.

Ready for your next step?Apply on the official website
Apply on WhatJobs ↗

Explore related searches

Current related jobs

Johns Hopkins Applied Physics Laboratory (APL)

WhatJobs

Reverse Engineer and AI Workflow Developer

laurel, MD

See pay details in description

Description Are you passionate about reverse engineering complex software, firmware, and hardware? Do you enjoy developing agentic AI workflo…

Listing review due 2026-10-06View job

Johns Hopkins Applied Physics Laboratory (APL)

WhatJobs

AFSIM Developer

laurel, MD

See pay details in description

Description Are you passionate about building high-fidelity simulations that shape the future of U.S. Space Force capabilities and force design …

Listing review due 2026-10-06View job

Johns Hopkins Applied Physics Laboratory (APL)

WhatJobs

Senior Software Engineer/Architect - TS SCI cleared

laurel, MD

See pay details in description

Description Are you passionate about building solutions for our greatest national security challenges? Are you searching for engaging work wi…

Listing review due 2026-10-06View job

Johns Hopkins Applied Physics Laboratory (APL)

WhatJobs

Senior Software Engineer

laurel, MD

See pay details in description

Description Are you passionate about building solutions for our greatest national security challenges? Are you searching for engaging work wi…

Listing review due 2026-10-06View job

Johns Hopkins Applied Physics Laboratory (APL)

WhatJobs

Senior Embedded Systems Developer

laurel, MD

See pay details in description

Description Do you love working on a motivated team to solve complex problems in innovative ways? Do you enjoy creating embedded prototypes i…

Listing review due 2026-10-06View job