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

Support for ARM based systems #8

Closed lumip closed 2 years ago

lumip commented 2 years ago

Due to its reliance on Intel SIMD intrinsics, the package can currently not be used on any ARM based systems like Apple's M1.

Support for such systems is desired. Therefore the CPU kernels need to be implemented also using ARM NEON intrinsics.