AMD GPU implementation - Githubissues

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

Closed lumip closed 1 year ago

lumip commented 1 year ago

So far we have only an implementation for Nvidia GPUs via CUDA. It would be good to also support GPU native execution on AMD hardware via ROCM.

lumip commented 1 year ago

included in v 2.0.0-rc.4