Skip to content

Commit

Permalink
Merge branch 'pstjohn/pyproject.toml' into 'main'
Browse files Browse the repository at this point in the history
expanding pyproject.toml definitions for uv

See merge request ADLR/megatron-lm!2084
  • Loading branch information
ko3n1g committed Sep 10, 2024
2 parents 3bdae05 + e93d566 commit b6887d3
Showing 1 changed file with 50 additions and 0 deletions.
50 changes: 50 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,56 @@ requires = [
"pybind11",
]

[project]
name = "megatron-core"
dynamic = ["dependencies", "version"]
description = "Megatron Core - a library for efficient and scalable training of transformer based models"
readme = "README.md"
license = {file = "LICENSE"}
authors = [{ name = "NVIDIA", email = "[email protected]" }]
maintainers = [{ name = "NVIDIA", email = "[email protected]" }]
keywords = [
"NLP",
"NLU",
"deep",
"gpu",
"language",
"learning",
"learning",
"machine",
"nvidia",
"pytorch",
"torch",
"transformer",
]
classifiers = [
"Development Status :: 5 - Production/Stable",
"Environment :: Console",
"Intended Audience :: Developers",
"Intended Audience :: Information Technology",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: BSD License",
"Natural Language :: English",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Image Recognition",
"Topic :: Scientific/Engineering :: Mathematics",
"Topic :: Scientific/Engineering",
"Topic :: Software Development :: Libraries :: Python Modules",
"Topic :: Software Development :: Libraries",
"Topic :: Utilities",
]

[tool.setuptools.dynamic]
dependencies = { file = ["megatron/core/requirements.txt"] }

[project.urls]
Download = "https://github.com/NVIDIA/Megatron-LM/releases"
Homepage = "https://github.com/NVIDIA/Megatron-LM/megatron/core"

[tool.isort]
profile = "black" # black-compatible
line_length = 100 # should match black parameters
Expand Down

0 comments on commit b6887d3

Please sign in to comment.