tachys

tachys is a JAX library for finding the ground states of quantum lattice models with neural-network wavefunctions. Build a Hamiltonian by composing spin and fermionic operators, use any Flax network as the wavefunction, and optimize it with variational Monte Carlo — the same code runs on a laptop and across a multi-GPU cluster. It also trains foundation models: a single network covering a whole family of Hamiltonians, rather than one run per coupling.

Documentation: https://tachys-dev.github.io/tachys/

Installation

CPU:

pip install tachys

CUDA (GPU):

pip install "tachys[cuda]"

This installs JAX with CUDA 12 support via the cuda optional dependency.

Install from source

git clone https://github.com/tachys-dev/tachys.git
cd tachys
pip install .

Development install

To install in editable mode (changes to the source are reflected immediately without reinstalling):

pip install -e ".[test]"          # CPU
pip install -e ".[cuda,test]"     # CUDA
pytest tests/

Authors

Riccardo Rende
Luciano Loris Viteritti

License

Licensed under the Apache License, Version 2.0.

Release files for tachys 0.1.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for tachys 0.1.1
File	Size	Uploaded
tachys-0.1.1.tar.gz	123.1 kB	Sep 24, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for tachys 0.1.1
File	Interpreter	ABI	Platform	Reset
tachys-0.1.1-py3-none-any.whl 129.2 kB Sep 24, 2026	Python 3	none	any	Details

Total release size: 252.3 kB

Release files / tachys-0.1.1.tar.gz

Download URL	tachys-0.1.1.tar.gz
Size	123.1 kB
Tags	Source
SHA-256 checksum How to use checksums	`b721d2c5ade1fb36aca24f21322d7a53c975d1640a51cbc4c2be6221b943da33`
BLAKE2b-256 checksum How to use checksums	`78baeda02b388e09794ebd8c2ef2b6e59a6e4c1fc181d5bc4152166eea2f410e`
Upload date	Sep 24, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/tachys-dev/tachys
Publishing commit: github.com/tachys-dev/tachys/tree/6a7646893e495ff63687c29c0cc489d23ba04de8

Workflow

Publishing configuration: github.com/tachys-dev/tachys/blob/6a7646893e495ff63687c29c0cc489d23ba04de8/.github/workflows/publish.yml
Publishing logs: github.com/tachys-dev/tachys/actions/runs/35939804177/attempts/1

Artifact

Subject: tachys-0.1.1.tar.gz
SHA-256 checksum: b721d2c5ade1fb36aca24f21322d7a53c975d1640a51cbc4c2be6221b943da33 Verifying attestations

Release files / tachys-0.1.1-py3-none-any.whl

Download URL	tachys-0.1.1-py3-none-any.whl
Size	129.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`d60e30ab8d25dc052a304ddf373f772136cd56a71ab487617dea649c9e5fcb4b`
BLAKE2b-256 checksum How to use checksums	`b40a9837926be4155b9deafd4dac65efa3b680719461adfba6b2fd2575084872`
Upload date	Sep 24, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/tachys-dev/tachys
Publishing commit: github.com/tachys-dev/tachys/tree/6a7646893e495ff63687c29c0cc489d23ba04de8

Workflow

Publishing configuration: github.com/tachys-dev/tachys/blob/6a7646893e495ff63687c29c0cc489d23ba04de8/.github/workflows/publish.yml
Publishing logs: github.com/tachys-dev/tachys/actions/runs/35939804177/attempts/1

Artifact

Subject: tachys-0.1.1-py3-none-any.whl
SHA-256 checksum: d60e30ab8d25dc052a304ddf373f772136cd56a71ab487617dea649c9e5fcb4b Verifying attestations