JaQMC#
JAX accelerated Quantum Monte Carlo
Modular by design
Swap and extend wavefunctions, samplers, estimators, and optimizers — from config or from code.
General-purpose
Atoms, molecules, and crystals. Extends naturally to other quantum systems.
Ready to use, ready to modify
Configure common neural network backbones and optimizers from the CLI or from code.
Built on JAX
Autodiff, JIT compilation, and multi-device parallelism for free.