README.md

jaxeigs

As we all known, until now, GPU version eigs is not supported in JAX. In this repo is a simple implementation of eigs in JAX by rescuing useful code in TensorNetwork by google.

Installation

Copy jaxeigs.py to your project.

Usage

See test.py for usage.