"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: Python, JAX, Machine Learning. Every listing is re-checked daily and closed roles are removed.

Showing 4 of 4 results

Sr. Software Engineer (Vehicle Engineering) Visa

Join SpaceX as a Senior Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $160k–$225k/yr Published 3 months ago

JAX Machine Learning reinforcement learning PyTorch TensorFlow

70% coding

Apply →

AI Software Engineer (Vehicle Engineering) Visa

Join SpaceX as an AI Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $125k–$175k/yr Published 3 months ago

JAX Machine Learning reinforcement learning PyTorch TensorFlow

Flexible on stack

Apply →

Sr. Software Engineer, Computer Vision Visa

SpaceX Hawthorne, CA $160k–$225k/yr Published 4 months ago

C++ Jupyter Grafana JAX Prometheus

Apply →

ML Engineer, Surrogate Modeling (Vehicle Engineering) Visa

Join SpaceX as an ML Engineer to develop AI surrogate models that enhance engineering simulations for launch vehicles and spacecraft.

SpaceX Hawthorne, CA $125k–$175k/yr Published 3 months ago

Surrogate Modeling Machine Learning NVIDIA PhysicsNemo FEA thermal analysis

Flexible on stack

Apply →