"jax" Jobs

4 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: PyTorch, Python, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 4 of 4 results

Member of Technical Staff, Research

Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.

Fireworks AI San Mateo Published 1 year ago

C++ JAX PyTorch TensorFlow Python

Flexible on stack

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity London Published 3 months ago

NVLink CuTe DSL Triton JAX CUTLASS

Apply →

Staff Machine Learning Engineer (Modeling), Support

Lead the development of machine learning initiatives to enhance customer support experiences at Block.

Block Seattle, WA, United States of America $276.8k–$415.2k/yr Published 1 month ago

JAX PyTorch TensorFlow Python

Flexible on stack

Apply →

Member of Technical Staff (AI Inference Engineer)

Perplexity San Francisco Published 3 months ago

NVLink CuTe DSL Nsight Systems Triton JAX

Apply →