JAX jobs

2 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.

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

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