Adapted `jax_ops.py` to handle the lowering interfaces changed from jax v0.4.16 onwards.

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 3 months ago

lumip commented 4 months ago

Also removed support and tests for Python 3.6 and 3.7, for which jaxlib packages are no longer available on pypi.

fixes #25

lumip commented 4 months ago

on hold: needs some further work to determine compatibility regressions to previous jax version