pyproject.toml

[project]
name = "thoad"
version = "0.1.0"
description = "thoad (Torch High Order Automatic Differentiation) is a lightweight reverse-mode autodifferentiation engine written entirely in Python that works over PyTorch’s computational graph to compute high order partial derivatives. Unlike PyTorch’s native autograd, which is limited to first-order native partial derivatives, thoad is able to performantly propagate arbitray-order derivatives throughout the graph, enabling more advanced gradient-based computations."
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "torch>=2.4.1",
]

[dependency-groups]
dev = [
    "pytest>=8.4.0",
    "jax>=0.7.0",
    "jaxlib>=0.7.0",
    "ipykernel>=6.30.1",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/thoad"]

[project.urls]
Homepage = "https://github.com/mntsx/thoad"
Issues = "https://github.com/mntsx/thoad/issues"