pyproject.toml

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "torch-shampoo"
version = "1.0.0"
dependencies = [
    "torch>=2.5.0",
]
requires-python = ">=3.10"
authors = [
  {name = "Hao-Jun Michael Shi"},
  {name = "Tsung-Hsien Lee"},
  {name = "Anna Cai"},
  {name = "Runa Eschenhagen"},
  {name = "Shintaro Iwasaki"},
  {name = "Ke Sang"},
  {name = "Wang Zhou"},
]
description = "For optimization algorithm research and development."
readme = "README.md"
keywords = ["deep-learning", "neural-networks", "optimization", "shampoo"]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

[project.optional-dependencies]
examples = [
  "torchvision>=0.15.0",
]

dev = [
  "torch-shampoo[examples]",
  "pre-commit",
  "ruff",
  "usort",
  "mypy",
  "expecttest",
]

[project.urls]
Repository = "https://github.com/facebookresearch/optimizers.git"
"Bug Tracker" = "https://github.com/facebookresearch/optimizers/issues"

[tool.setuptools]
include-package-data = false
py-modules = [
    "commons",
    "matrix_functions",
    "matrix_functions_types",
    "optimizer_modules",
]

[tool.setuptools.packages.find]
include = ["distributed_shampoo*"]
exclude = ["*examples", "*tests"]

[tool.usort]
first_party_detection = false