rkarur
Follow
Stars
7
results
for source starred repositories
written in Python
Clear filter
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).