anchor.diff:      "tests/"
     ];
   
  +  # See https://github.com/google/jax/issues/9705.
  +  disabledTests = lib.optionals usingMKL [ "test_custom_root_with_aux" ];
  +
     pythonImportsCheck = [
       "jax"
     ];
  
positive.code: , typing-extensions
  }:
  
  buildPythonPackage rec {
    pname = "jax";
    version = "0.3.1";
  
negative.code: { lib
  , absl-py
  , blas
  , buildPythonPackage
  , fetchFromGitHub
  , jaxlib
  , lapack
  , numpy
  , opt-einsum
  , pytestCheckHook
  