Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools,
# dependencies absl-py, jax, numpy,
# tests pytestCheckHook,}:
buildPythonPackage (finalAttrs: { pname = "jmp"; version = "0.0.4"; pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "deepmind"; repo = "jmp"; tag = "v${finalAttrs.version}"; hash = "sha256-+PefZU1209vvf1SfF8DXiTvKYEnZ4y8iiIr8yKikx9Y="; };
build-system = [ setuptools ];
dependencies = [ numpy # Not listed in `dependencies` but in practice is necessary to import jmp absl-py jax ];
pythonImportsCheck = [ "jmp" ];
nativeCheckInputs = [ pytestCheckHook ];
meta = { description = "This library implements support for mixed precision training in JAX"; homepage = "https://github.com/deepmind/jmp"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ ndl ]; };})