jax

524d4254 - update xla link, fixes #1248