fix jax-doclets