v0.2.6

What's Changed

• Regression fix and some tests by @ASKabalan in #63
• Bumping to 0.2.6 by @ASKabalan in #64

Full Changelog: v0.2.5...v0.2.6

v0.2.6 is the most complete version of jaxDecomp
Distributed FFTs now support jacrev, jacfwd and vmap

v0.2.5

What's Changed

• 60 remove shardedarray by @ASKabalan in #61
• Updating readme and bumping version by @ASKabalan in #62

Full Changelog: v0.2.4...v0.2.5

v0.2.4

What's Changed

• 52 implement shardmapped ops by @ASKabalan in #53
• update cuDecomp version by @ASKabalan in #55
• PFFT returns ShardedArray if input is ShardedArray by @ASKabalan in #56
• bump version to 0.2.4 by @ASKabalan in #57

Full Changelog: v0.2.3...v0.2.4

v0.2.3

What's Changed

• bump jax version in dependencies by @ASKabalan in #51
• Fix gradient issue with PFFT3D by @ASKabalan in #49

Full Changelog: v0.2.2...v0.2.3

v0.2.2

What's Changed

• bumping version by @ASKabalan in #47

Releasing a new version with cleaned up build wheels that works for python 3.10 3.11 and 3.12

Full Changelog: v0.2.1...v0.2.2

v0.2.1

What's Changed

• Push to pypi the sequel by @ASKabalan in #32
• Push to pypi the sequel by @ASKabalan in #33
• publish to pypi on release by @ASKabalan in #34
• remove condition for pypi release by @ASKabalan in #35
• Update github-deploy.yml by @ASKabalan in #36
• 38 fix issue with cyclic import by @ASKabalan in #39
• No testing on release by @ASKabalan in #40
• expand test matrix by @ASKabalan in #41
• update python wheel api in toml by @ASKabalan in #44
• Loosen JAX version constraint by @ASKabalan in #45
• only build for x64 by @ASKabalan in #46

Full Changelog: v0.2.0...v0.2.1

v0.2.0

What's Changed

• One transpositions for slabs by @ASKabalan in #27
• update install in README by @ASKabalan in #29
• Implement jax versions by @ASKabalan in #30
• Non contiguous transpose axis by @ASKabalan in #28
• Docstrings by @ASKabalan in #31

Full Changelog: v0.1.0...v0.2.0

v0.1.0 first stable release of jaxDecomp

What's Changed

• Remove custom assignment of gpu by @EiffL in #14
• Add a clang-format pre-commit and workflow by @ASKabalan in #16
• Fix 3D FFT by @EiffL in #12
• Implement Transpose primitives by @ASKabalan in #17
• fix: some CMake fixes by @henryiii in #21
• Fix slice unpad by @ASKabalan in #18
• Clean up for JOSS paper by @ASKabalan in #19
• [Packaging] Migrating to scikit-build core by @ASKabalan in #22
• Proposal for simplified demo by @EiffL in #24
• Joss paper of jaxDecomp by @ASKabalan in #20
• Fix halo again by @ASKabalan in #23
• update changelog by @ASKabalan in #25

New Contributors

• @henryiii made their first contribution in #21

Full Changelog: v0.0.1rc2...v0.1.0

Updated version compatible with jax v0.4.xx

v0.0.1rc2

Merge pull request #7 from DifferentiableUniverseInitiative/update-ja…

First release candidate for initial API

Implements initial set of functionalities, in particular gives access to 3D FFT and halo exchange.