Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 70 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071{ lib, buildPythonPackage, fetchPypi, jax, jaxlib, multipledispatch, numpy, pytestCheckHook, pythonOlder, tensorflow-probability, tqdm}:
buildPythonPackage rec { pname = "numpyro"; version = "0.11.0"; format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchPypi { inherit version pname; hash = "sha256-01fdGgFZ+G1FwjNwitM6PT1TQx0FtLvs4dBorkFoqo4="; };
propagatedBuildInputs = [ jax jaxlib numpy multipledispatch tqdm ];
nativeCheckInputs = [ tensorflow-probability pytestCheckHook ];
pythonImportsCheck = [ "numpyro" ];
disabledTests = [ # AssertionError due to tolerance issues "test_beta_binomial_log_prob" "test_collapse_beta" "test_cpu" "test_gamma_poisson" "test_gof" "test_hpdi" "test_kl_dirichlet_dirichlet" "test_kl_univariate" "test_mean_var" # Tests want to download data "data_load" "test_jsb_chorales" # RuntimeWarning: overflow encountered in cast "test_zero_inflated_logits_probs_agree" # NameError: unbound axis name: _provenance "test_model_transformation" ];
meta = with lib; { description = "Library for probabilistic programming with NumPy"; homepage = "https://num.pyro.ai/"; changelog = "https://github.com/pyro-ppl/numpyro/releases/tag/${version}"; license = licenses.asl20; maintainers = with maintainers; [ fab ]; };}