jax

2e34dbc1 - update travis to match min jaxlib version