Documentation#
Homepage
Looking for the project overview? Head back to the tachys homepage.
Benchmarks
Reference variational energies for frustrated lattice models — pick a model, a lattice geometry and a size on the benchmarks page.
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.
Installation#
pip install tachys # CPU
pip install "tachys[cuda]" # GPU, with JAX for CUDA 12
Getting Started
Documentation
Additional Resources