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, buildPythonPackage, pythonOlder, fetchFromGitHub, pytest-xdist, pytestCheckHook, absl-py, cvxpy, jax, jaxlib, matplotlib, numpy, optax, scipy, scikit-learn}:
buildPythonPackage rec { pname = "jaxopt"; version = "0.8.3"; format = "setuptools";
disabled = pythonOlder "3.8";
src = fetchFromGitHub { owner = "google"; repo = "jaxopt"; rev = "refs/tags/jaxopt-v${version}"; hash = "sha256-T/BHSnuk3IRuLkBj3Hvb/tFIb7Au25jjQtvwL28OU1U="; };
propagatedBuildInputs = [ absl-py jax jaxlib matplotlib numpy scipy ];
nativeCheckInputs = [ pytest-xdist pytestCheckHook cvxpy optax scikit-learn ];
pythonImportsCheck = [ "jaxopt" "jaxopt.implicit_diff" "jaxopt.linear_solve" "jaxopt.loss" "jaxopt.tree_util" ];
meta = with lib; { homepage = "https://jaxopt.github.io"; description = "Hardware accelerated, batchable and differentiable optimizers in JAX"; changelog = "https://github.com/google/jaxopt/releases/tag/jaxopt-v${version}"; license = licenses.asl20; maintainers = with maintainers; [ bcdarwin ]; };}