From a2808548304ec9485b95e4a4f0f16bd01b42cac8 Mon Sep 17 00:00:00 2001 From: Johan Herland Date: Mon, 15 Jun 2026 10:24:46 +0200 Subject: [PATCH] python3Packages.liger-kernel: init at 0.8.0 --- .../python-modules/liger-kernel/default.nix | 45 +++++++++++++++++++ pkgs/top-level/python-packages.nix | 2 + 2 files changed, 47 insertions(+) create mode 100644 pkgs/development/python-modules/liger-kernel/default.nix diff --git a/pkgs/development/python-modules/liger-kernel/default.nix b/pkgs/development/python-modules/liger-kernel/default.nix new file mode 100644 index 000000000000..8454e6933a35 --- /dev/null +++ b/pkgs/development/python-modules/liger-kernel/default.nix @@ -0,0 +1,45 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + setuptools-scm, + torch, + triton, +}: + +buildPythonPackage (finalAttrs: { + pname = "liger-kernel"; + version = "0.8.0"; + pyproject = true; + + src = fetchFromGitHub { + owner = "linkedin"; + repo = "liger-kernel"; + tag = "v${finalAttrs.version}"; + hash = "sha256-MAJ0goMxi8qA+ODd/kLIO7hp4dmNIOPLpSdkrm7hnnQ="; + }; + + 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 ]; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 11071cbaf5cc..44e64630d788 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9102,6 +9102,8 @@ self: super: with self; { lifx-emulator-core = callPackage ../development/python-modules/lifx-emulator-core { }; + liger-kernel = callPackage ../development/python-modules/liger-kernel { }; + lightgbm = callPackage ../development/python-modules/lightgbm { }; lightify = callPackage ../development/python-modules/lightify { };