Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546{ lib, buildPythonPackage, fetchFromGitHub, setuptools, setuptools-scm, torch, triton,}:
buildPythonPackage (finalAttrs: { pname = "liger-kernel"; version = "0.8.5"; pyproject = true;
src = fetchFromGitHub { owner = "linkedin"; repo = "liger-kernel"; tag = "v${finalAttrs.version}"; hash = "sha256-Gsdp+3+r7Vlx+bJurjKZSM5ZK7ucsR2TNWvcMImg7uw="; };
build-system = [ setuptools setuptools-scm ];
dependencies = [ torch triton ];
# Requires NVIDIA driver. doCheck = false;
pythonImportsCheck = [ "liger_kernel" ];
meta = { description = "Efficient Triton Kernels for LLM Training"; homepage = "https://github.com/linkedin/liger-kernel"; changelog = "https://github.com/linkedin/liger-kernel/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.bsd3; maintainers = with lib.maintainers; [ jherland ]; };})