Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
900 B · 49 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950{ lib, fetchFromGitHub, buildPythonPackage, jax, jaxlib, numpy, parameterized, pillow, scipy, tensorboard}:
buildPythonPackage rec { pname = "objax"; version = "1.6.0";
src = fetchFromGitHub { owner = "google"; repo = "objax"; rev = "v${version}"; hash = "sha256-/6tZxVDe/3C53Re14odU9VA3mKvSj9X3/xt6bHFLHwQ="; };
# Avoid propagating the dependency on `jaxlib`, see # https://github.com/NixOS/nixpkgs/issues/156767 buildInputs = [ jaxlib ];
propagatedBuildInputs = [ jax numpy parameterized pillow scipy tensorboard ];
pythonImportsCheck = [ "objax" ];
meta = with lib; { description = "Objax is a machine learning framework that provides an Object Oriented layer for JAX."; homepage = "https://github.com/google/objax"; license = licenses.asl20; maintainers = with maintainers; [ ndl ]; };}