pyproject.toml

# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

[project]
name = "warp-nn"
version = "0.4.0"
requires-python = ">=3.10"
description = "Neural network library for NVIDIA Warp"
license = { text = "Apache-2.0" }
readme = "README.md"
authors = [
  {name = "NVIDIA Corporation"},
]
keywords = ["machine-learning", "neural-network", "nvidia", "warp"]
classifiers = [
  "Environment :: GPU :: NVIDIA CUDA",
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",
  "Programming Language :: Python :: 3",
  "Operating System :: OS Independent",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
  "warp-lang>=1.15.0",
]

[project.urls]
Documentation = "https://nvidia.github.io/warp-nn"
Repository = "https://github.com/NVIDIA/warp-nn"
Issues = "https://github.com/NVIDIA/warp-nn/issues"
Changelog = "https://github.com/NVIDIA/warp-nn/blob/main/CHANGELOG.md"

[project.optional-dependencies]
onnx = [
  "onnx>=1.16.0",  # no `onnxruntime`, just protobuf parser. Inference runs on Warp kernels
]
tests = [
  "hypothesis",
  "onnx>=1.16.0",
  "pytest",
  "pytest-cov",
  "torch",
]
docs = [
  "myst-parser",
  "nvidia-sphinx-theme",
  "sphinx-autobuild",
  "sphinx-copybutton",
]

# ---------------------------------------------------------------------

[tool.setuptools.packages.find]
where = ["."]
include = ["warp_nn*"]


[tool.black]
line-length = 120
extend-exclude = """
(
  ^/docs
)
"""


[tool.codespell]
# run: codespell <folder>
skip = "./docs/source/_static,./docs/_build,pyproject.toml"
quiet-level = 3
count = true


[tool.isort]
profile = "black"
line_length = 120
lines_after_imports = 2
known_test = [
  "warnings",
  "hypothesis",
  "pytest",
]
known_annotation = ["typing"]
known_framework = [
  "numpy",
  "warp",
]
sections = [
  "FUTURE",
  "ANNOTATION",
  "TEST",
  "STDLIB",
  "THIRDPARTY",
  "FRAMEWORK",
  "FIRSTPARTY",
  "LOCALFOLDER",
]
no_lines_before = "THIRDPARTY"
skip = ["docs"]