Update jax.js

Typo happens.