rkarur
Follow
Stars
Use Python for lattice QCD research based on the QUDA library.
QUDA is a library for performing calculations in lattice QCD on GPUs.
An Aspiring Drop-In Replacement for NumPy at Scale
A Tutorial for JAX Multi controller distribution
Distributed JAX at NERSC (demo).
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Add a tqdm progress bar to your JAX scans and loops.
A loosely related collection of lattice field theory code.
MCHMC: sampler from an arbitrary differentiable distribution