Role directory

JAX jobs

3 current, verified opportunities. Compare the requirements here, then complete applications on the partner platform.

Current opportunities

Verified listing details with no HumanitApp account or application fee.

Language / Remote — United States

GPU Kernel Expert

Evaluate the quality, correctness, and completeness of GPU/accelerator kernel development tasks used to train and evaluate a frontier AI lab's models. You'll assess numerical correctness, performance-benchmarking fairness, task scoping,...

$70 - $90 / hourVerified / Application completed externally
Code / Remote

LLM Research Scientist (Pre-training & Computer Vision & Adversarial Robustness)

We're looking for experienced machine learning researchers with hands-on experience training and improving deep learning models end-to-end, across vision and language. You'll work on well-scoped empirical open-ended ML research problems. —...

$100 - $120 / hourVerified / Application completed externally
Code / Canada, UK, US Remote

MLOps Engineer, LLM Systems (Serving, GPU Kernels, Profiling)

Join a leading AI lab's cutting-edge GenAI team to be at the core of the AI revolution, where your expertise fuels the development of the most advanced Large Language Models. — 1. Overview — Join a leading AI lab's cutting-edge GenAI team...

$90 - $120 / hourVerified / Application completed externally