Commit 309c92d4 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python310Packages.jax: 0.4.14 -> 0.4.15

parent ad9c1e95

pkgs/development/python-modules/jax/default.nix

+6 −2
Original line number Diff line number Diff line
@@ -27,7 +27,7 @@ let 
in

buildPythonPackage rec {

  pname = "jax";

  version = "0.4.14";

  version = "0.4.16";

  format = "pyproject";



  disabled = pythonOlder "3.9";
@@ -37,13 +37,17 @@ buildPythonPackage rec { 
    repo = pname;

    # google/jax contains tags for jax and jaxlib. Only use jax tags!

    rev = "refs/tags/${pname}-v${version}";

    hash = "sha256-0KnILQkahSiA1uuyT+kgy1XaCcZ3cpx1q114e2pecvg=";

    hash = "sha256-q+8CXGxK8JX0bUMK4KJB3qV/EaLHg68D1B5UrtRz0Eg=";

  };



  nativeBuildInputs = [

    setuptools

  ];



  # The version is automatically set to ".dev" if this variable is not set.

  # https://github.com/google/jax/commit/e01f2617b85c5bdffc5ffb60b3d8d8ca9519a1f3

  JAX_RELEASE = "1";



  # jaxlib is _not_ included in propagatedBuildInputs because there are

  # different versions of jaxlib depending on the desired target hardware. The

  # JAX project ships separate wheels for CPU, GPU, and TPU.