Releases: DPBayes/jax-chacha-prng
Releases · DPBayes/jax-chacha-prng
1.4.3
2.0.0-rc.4
Necessary changes in registrating GPU kernels to retain compatibility with most recent jax releases
Added GPU kernels and support for AMD ROCM compilation.
Breaking:
- drops support for all jax prior to 0.4.1
- reverts earlier v2.0.0-rc.3 and prior changes (new rng split routines - to be released in v3)
1.4.2
1.4.1
- Fix: native module not built for correct Python verison for MacOS.
- Fix: Wheels for MacOS no longer build with OpenMP to avoid complicatons with conda environments.
1.4.0.post1
Removed local version suffix to differentiate cpu and cuda builds from wheels for upload to pypi, which doesn't accept those.
No changes to software from prior 1.4.0 release.
1.4.0
fixing compatibility tests
1.3.0
- Fix: OpenMP is no longer a strict requirement for installation.
- Added: chacha.native.openmp_accelerated, returns True if CPU kernels are parallelised using OpenMP.
- Added: Generic (not hardware-specific) CPU kernels as well as vectorized CPU kernels for ARM CPUs.
- Added: support for jax up to v0.3.25
- Added: support for Python v3.11
Note: jax-chacha-prng is now also available from PyPi.org for easy installation via pip.
1.4.0-rc.1
Dropped support for jax before v0.2.17
1.2.1
- Added: support for jax up to v0.3.23.
2.0.0-rc.3
v2.0.0-rc.3