CUDA builds for PyPI - 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

current wheels uploaded to PyPI are CPU only; for higher user convenience, they should support CUDA