An Aspiring Drop-In Replacement for NumPy at Scale

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Add a tqdm progress bar to your JAX scans and loops.

Use Python for lattice QCD research based on the QUDA library.

A Tutorial for JAX Multi controller distribution

Distributed JAX at NERSC (demo).

Python interface to lattice/QUDA