Metadata-Version: 2.4
Name: tachys
Version: 0.1.1
Summary: Neural quantum states in JAX: variational Monte Carlo for quantum lattice models
Author: Riccardo Rende, Luciano Loris Viteritti
License-Expression: Apache-2.0
Project-URL: Homepage, https://github.com/tachys-dev/tachys
Project-URL: Documentation, https://tachys-dev.github.io/tachys/
Project-URL: Repository, https://github.com/tachys-dev/tachys
Project-URL: Issues, https://github.com/tachys-dev/tachys/issues
Keywords: jax,neural quantum states,variational monte carlo,quantum many-body
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Topic :: Scientific/Engineering :: Physics
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: NOTICE
Requires-Dist: jax==0.7.2
Requires-Dist: flax==0.12.0
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: einops
Requires-Dist: numba
Requires-Dist: optax
Requires-Dist: orbax-checkpoint
Provides-Extra: cuda
Requires-Dist: jax[cuda12]==0.7.2; extra == "cuda"
Provides-Extra: wandb
Requires-Dist: wandb; extra == "wandb"
Requires-Dist: jsonargparse; extra == "wandb"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Provides-Extra: docs
Requires-Dist: sphinx>=7.2; extra == "docs"
Requires-Dist: pydata-sphinx-theme>=0.15; extra == "docs"
Requires-Dist: myst-parser>=2.0; extra == "docs"
Requires-Dist: sphinx-copybutton>=0.5; extra == "docs"
Requires-Dist: sphinx-design>=0.5; extra == "docs"
Dynamic: license-file

<p align="center">
  <img src="https://raw.githubusercontent.com/tachys-dev/tachys/main/docs/_static/logo.svg" width="285" height="48" alt="tachys logo">
</p>

# 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:**
```bash
pip install tachys
```

**CUDA (GPU):**
```bash
pip install "tachys[cuda]"
```

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

### Install from source

```bash
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):

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

## Authors

- Riccardo Rende
- Luciano Loris Viteritti

## License

Copyright 2026 The Simons Foundation, Inc.

Licensed under the [Apache License, Version 2.0](https://github.com/tachys-dev/tachys/blob/main/LICENSE).
