File size: 1,101 Bytes
71a70fb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | [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*"]
|