Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.7 kB · 74 lines
Nix
at llvm-cleanup
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub, hatchling, jax, jaxlib, jaxtyping, typing-extensions, beartype, optax, pytest-xdist, pytestCheckHook,}:
buildPythonPackage rec { pname = "equinox"; version = "0.11.4"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "patrick-kidger"; repo = "equinox"; rev = "refs/tags/v${version}"; hash = "sha256-3OwHND1YEdg/SppqiB7pCdp6v+lYwTbtX07tmyEMWDo="; };
nativeBuildInputs = [ hatchling ];
propagatedBuildInputs = [ jax jaxlib jaxtyping typing-extensions ];
nativeCheckInputs = [ beartype optax pytest-xdist pytestCheckHook ];
pythonImportsCheck = [ "equinox" ];
disabledTests = [ # For simplicity, JAX has removed its internal frames from the traceback of the following exception. # https://github.com/patrick-kidger/equinox/issues/716 "test_abstract" "test_complicated" "test_grad" "test_jvp" "test_mlp" "test_num_traces" "test_pytree_in" "test_simple" "test_vmap"
# AssertionError: assert 'foo:\n pri...pe=float32)\n' == 'foo:\n pri...pe=float32)\n' # Also reported in patrick-kidger/equinox#716 "test_backward_nan" ];
meta = with lib; { description = "A JAX library based around a simple idea: represent parameterised functions (such as neural networks) as PyTrees"; changelog = "https://github.com/patrick-kidger/equinox/releases/tag/v${version}"; homepage = "https://github.com/patrick-kidger/equinox"; license = licenses.asl20; maintainers = with maintainers; [ GaetanLepage ]; };}