"jax" Jobs

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

Showing 2 of 2 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 →

MTS, Research Engineer

Join Fireworks AI as a Research Engineer to tackle open-ended research problems and build scalable training infrastructure.

Fireworks AI San Mateo Published 2 weeks ago

C++ JAX CUDA Rust PyTorch

Flexible on stack

Apply →