--- a +++ b/sub-packages/bionemo-llm/pyproject.toml @@ -0,0 +1,44 @@ +[build-system] +requires = ["setuptools>=64", "wheel"] +build-backend = "setuptools.build_meta" + +[project] +name = "bionemo-llm" +readme = "README.md" +description = "BioNeMo Large Language Model Components using NeMo and Megatron" +authors = [{ name = "BioNeMo Team", email = "bionemofeedback@nvidia.com" }] +requires-python = ">=3.10" +license = { file = "LICENSE" } +dynamic = ["version"] +dependencies = [ + # bionemo sub-packages + 'bionemo-core', + # external + 'lightning>=2.2.1', + 'megatron-core', + 'nemo_toolkit[nlp]>=2.2.1', + 'nemo-run', + 'hatchling', +] + +[project.optional-dependencies] +test = [ + 'bionemo-testing' +] +te = [ + # TE & Apex need to be installed after PyTorch, NVCC, and CUDA. + # TODO(@pstjohn, @cspades): Figure out how to do this without post-installation. + 'transformer_engine[pytorch]' +] + +[tool.setuptools.packages.find] +where = ["src"] +include = ["bionemo.*"] +namespaces = true +exclude = ["test*."] + +[tool.setuptools.dynamic] +version = { file = "VERSION" } + +[tool.uv] +cache-keys = [{ git = true }]