Expose hidden_axes via jax namespace as public API. Also mention it as a workaround for primitives we don't support yet.#25970