"jax" Jobs

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

Showing 3 of 3 results

Forward Deployed Engineer

Join Goodfire as a Forward Deployed Engineer to integrate AI interpretability solutions with strategic partners.

Goodfire San Francisco, CA $200k–$325k/yr Published 5 months ago

JAX PyTorch Python

Flexible on stack 70% coding

Apply →

Machine Learning Engineer Hybrid

Join Goodfire as a Machine Learning Engineer to build interpretable AI systems with a world-class team.

Goodfire San Francisco, CA & New York, NY $200k–$400k/yr Published 9 months ago

JAX distributed systems PyTorch Python

Apply →

Head of Engineering, Platform

Join Goodfire as the Head of Engineering to lead the development of AI interpretability tools with a world-class team.

Goodfire San Francisco, CA $300k–$450k/yr Published 8 months ago

JAX distributed systems large-scale model training PyTorch ML infrastructure

Flexible on stack

Apply →