This prevents low-level errors recently introduced by newer versions of
`jaxlib`.