Relax tolerances slightly for MKL#9706

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom

samuela:samuela/mkl-fixsamuela/jax:samuela/mkl-fixCopy head branch name to clipboard