Something went wrong. Try again.
lol
Something went wrong. Try again.
949 B · 43 lines
Nix
at master
1234567891011121314151617181920212223242526272829303132333435363738394041424344{ buildPythonPackage, einops, fetchFromGitHub, jax, jaxlib, lib, setuptools,}:
buildPythonPackage rec { pname = "augmax"; version = "0.3.3"; pyproject = true;
src = fetchFromGitHub { owner = "khdlr"; repo = "augmax"; tag = "v${version}"; hash = "sha256-FXgkhZEAR1Y2LvVvV+IWMSQDWrLulLDsSKKuw4ER5wg="; };
build-system = [ setuptools ];
dependencies = [ einops jax ];
# augmax does not have any tests at the time of writing (2022-02-19), but # jaxlib is necessary for the pythonImportsCheckPhase. nativeCheckInputs = [ jaxlib ];
pythonImportsCheck = [ "augmax" ];
meta = { description = "Efficiently Composable Data Augmentation on the GPU with Jax"; homepage = "https://github.com/khdlr/augmax"; changelog = "https://github.com/khdlr/augmax/blob/v${version}/CHANGELOG.md"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ samuela ]; };}