Releases: DifferentiableUniverseInitiative/jaxDecomp
Releases · DifferentiableUniverseInitiative/jaxDecomp
Updated version compatible with jax v0.4.xx
Merge pull request #7 from DifferentiableUniverseInitiative/update-ja… …x-array-api Update jax array api
First release candidate for initial API
Implements initial set of functionalities, in particular gives access to 3D FFT and halo exchange.