Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.2 kB · 64 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465{ lib, absl-py, buildPythonPackage, chex, fetchFromGitHub, jaxlib, numpy, callPackage, pythonOlder}:
buildPythonPackage rec { pname = "optax"; version = "0.1.7"; format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "deepmind"; repo = pname; rev = "refs/tags/v${version}"; hash = "sha256-zSMJxagPe2rkhrawJ+TWXUzk6V58IY6MhWmEqLVtOoA="; };
outputs = [ "out" "testsout" ];
buildInputs = [ jaxlib ];
propagatedBuildInputs = [ absl-py chex numpy ];
postInstall = '' mkdir $testsout cp -R examples $testsout/examples '';
pythonImportsCheck = [ "optax" ];
# check in passthru.tests.pytest to escape infinite recursion with flax doCheck = false;
passthru.tests = { pytest = callPackage ./tests.nix { }; };
meta = with lib; { description = "Gradient processing and optimization library for JAX"; homepage = "https://github.com/deepmind/optax"; changelog = "https://github.com/deepmind/optax/releases/tag/v${version}"; license = licenses.asl20; maintainers = with maintainers; [ ndl ]; };}