"jax" Jobs
8 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, Python. Every listing is re-checked daily and closed roles are removed.
Showing 8 of 8 results
Lead the unification of large datasets to power generative audio models in a fast-moving team at Udio.
Join AssemblyAI as a Senior Research Engineer to enhance large-scale distributed training and inference systems in Voice AI.
Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.
Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.
Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.