Senior ML Research Engineer - Inference & GPU Kernels

AfterQuery; Tempe (AZ); Remote; USD 138,000 - 234,000; Full time

Benefits offered by this job

Frontier AI work

Fully remote

Competitive pay

Job summary

AfterQuery seeks an ML Research Engineer focusing on inference and GPU kernels for remote, contract work. You will architect challenging evaluation problems, craft reference solutions, and judge AI outputs with rigor and domain insight.

The role emphasizes research, not production engineering, and invites candidates with deep expertise in frontier AI models and a strong track record in related publications or projects.

Qualifications

• Currently or recently active in the field — hands-on, not purely academic-adjacent
• Strong written communication — you'll be authoring technical explanations and structured feedback, not just doing the work itself
• Comfortable with independent, asynchronous, remote work
• Meets the experience and education bar described in Full Description

Responsibilities

• Design realistic technical scenarios and problem sets within your track
• Author expert-level reference solutions and grading rubrics
• Evaluate AI-generated outputs for correctness, depth, and domain judgment

Skills

Research experience

Written communication

Independent work

Asynchronous collaboration

Education

PhD

Master's with strong research record

Tools

SGLang

vLLM

Mamba2

TensorRT-LLM

Job description

AfterQuery seeks an ML Research Engineer focusing on inference and GPU kernels for remote, contract work. You will architect challenging evaluation problems, craft reference solutions, and judge AI outputs with rigor and domain insight.

The role emphasizes research, not production engineering, and invites candidates with deep expertise in frontier AI models and a strong track record in related publications or projects.