Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.1 kB · 66 lines
Nix
at litex
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub, pytestCheckHook, absl-py, cvxpy, jax, jaxlib, matplotlib, numpy, optax, scipy, scikit-learn}:
buildPythonPackage rec { pname = "jaxopt"; version = "0.5.5"; format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "google"; repo = pname; rev = "refs/tags/${pname}-v${version}"; hash = "sha256-WOsr/Dvguu9/qX6+LMlAKM3EANtYPtDu8Uo2157+bs0="; };
propagatedBuildInputs = [ absl-py jax jaxlib matplotlib numpy scipy ];
nativeCheckInputs = [ pytestCheckHook cvxpy optax scikit-learn ];
pythonImportsCheck = [ "jaxopt" "jaxopt.implicit_diff" "jaxopt.linear_solve" "jaxopt.loss" "jaxopt.tree_util" ];
disabledTests = [ # Stack frame issue "test_bisect" ];
meta = with lib; { homepage = "https://jaxopt.github.io"; description = "Hardware accelerated, batchable and differentiable optimizers in JAX"; license = licenses.asl20; maintainers = with maintainers; [ bcdarwin ]; };}