Jack's Jax Utilities

A collection of utilities I've written for jax.

Currently provides:

• LOBPCG implementation for optionally sparse eigendecomposition.
• vjp for partial eigendecomposition, with optionally sparse inputs.

Pre-commit

This package uses pre-commit to ensure commits meet minimum criteria. To Install, use

pip install pre-commit
pre-commit install

This will ensure git hooks are run before each commit. While it is not advised to do so, you can skip these hooks with

git commit --no-verify -m "commit message"