anchor.diff:    checkInputs = [
       jaxlib
       pytestCheckHook
  +    pytest-xdist
     ];
   
     # NOTE: Don't run the tests in the expiremental directory as they require flax
  
positive.code: 
  buildPythonPackage rec {
    pname = "jax";
    version = "0.2.28";
    format = "setuptools";
  
    disabled = pythonOlder "3.7";
  
negative.code:     owner = "google";
      repo = pname;
      rev = "${pname}-v${version}";
      sha256 = "0ndpngx5k6lf6jqjck82bbp0gs943z0wh7vs9gwbyk2bw0da7w72";
    };
  
    patches = [
  