[build-system]
requires = ["setuptools>=77", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "hamiltonzero"
version = "0.1.0"
description = "Compiled neural wavefunctions for quantum spin systems"
readme = "README.md"
requires-python = ">=3.12"
license = "Apache-2.0 AND MIT"
license-files = [
    "LICENSE",
    "NOTICE",
    "THIRD_PARTY_NOTICES.md",
    "third_party/folx/LICENSE",
    "third_party/jax/LICENSE",
    "third_party/kfac_jax/LICENSE",
]
dependencies = [
    "absl-py>=2.5",
    "distrax>=0.1.9",
    "dm-tree>=0.1.10",
    "equinox==0.13.6",
    "immutabledict>=4.3",
    "jax @ git+https://github.com/TakeOver/jax.git@79f82535b15a444516d4a5e2beb71d283665b2ff",
    "jaxlib==0.11.0",
    "jaxtyping==0.3.9",
    "networkx>=3.5",
    "numpy>=2.3",
    "optax>=0.2.8",
    "packaging>=26",
    "typing-extensions>=4.15",
]

[project.optional-dependencies]
notebooks = [
    "jupyterlab>=4.4",
    "matplotlib>=3.10",
]

[project.scripts]
hamiltonzero = "hamiltonzero.cli:main"

[tool.setuptools.packages.find]
where = ["src"]
include = ["hamiltonzero*", "kfac_jax*"]