issues
search
DPBayes
/
jax-chacha-prng
A cryptographically-secure pseudo-random number generator for JAX based on the 20 round ChaCha cipher.
12
stars
1
forks
source link
v2
#24
Closed
lumip
closed
1 year ago
lumip
commented
1 year ago
AMD ROCM kernels
Necessary updates to GPU primitive registration for compatibility with most recent jax versions
Dropped support for JAX < 0.4.1 and Python < 3.8