Recent Releases of jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.3.1
What's Changed
- Support nested sharding by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/113
- Doc and isntallation updates by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/114
- update JaxTimer and jax hpc profiler dependencies by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/115
- Update paper plots by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/117
- Update figures by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/118
- update guidelines by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/119
- JOSS review by @diehlpk in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/120
- Bump to v0.3.1 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/123
New Contributors
- @diehlpk made their first contribution in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/120
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.3.0...v0.3.1
- Python
Published by ASKabalan 13 days ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.3.0
What's Changed
- Update for joss review 2 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/98
- refactor: improve validation and error handling in cuDecomp backend by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/102
- Update cudecomp cpp by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/103
- update docs and README to use AxisType.Auto in mesh creation by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/104
- Migrating to ffi by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/105
- Update docs by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/106
- Fix bug with transpose and halo exchange lowering by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/107
- Add a better benchmark script by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/108
- jax.tools.colabtpu.setuptpu() no longer needed by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/109
- Update CITATION by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/110
- Bump version to 0.3.0 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/111
- Fix build wheel action by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/112
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.9...v0.3.0
- Python
Published by ASKabalan 4 months ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.9
What's Changed
- Updating README.md with install instructions for cuDecomp backend by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/91
- Change test paramters for better reading and add pytest-cov to test matrix by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/92
- Add new CITATION.cff file by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/93
- Update jaxDecomp to fix errors and deprecations introduced in jax 0.8.0 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/94
- Bump version to v0.2.9 by @Copilot in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/96
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.8...v0.2.9
- Python
Published by ASKabalan 7 months ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.8
What's Changed
- Joss paper by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/26
- update docs by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/70
- add docs badge by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/71
- Update and deploy READTHEDOCS by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/72
- Update Latex extension for docs by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/73
- Fix error in FFT normalization for large mesh by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/76
- minor edits to the JOSS paper by @EiffL in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/75
- Continue Trimming paper by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/77
- Some more trimming by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/78
- More trimming by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/79
- Final DOI and trimming by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/80
- Updating JOSS paper by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/81
- Update links to go to documentation site instead of md files by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/85
- Implement partitionning using shardy by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/86
- bump to v0.2.8 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/88
- Shardy Partionner must not be used prior to jax 0.7.0 including 0.7.0 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/90
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.7...v0.2.8
- Python
Published by ASKabalan 8 months ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.7
What's Changed
- use FFI calling instead of xla client by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/67
- bump to 0.2.7 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/68
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.6...v0.2.7
- Python
Published by ASKabalan about 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.6
What's Changed
- Regression fix and some tests by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/63
- Bumping to 0.2.6 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/64
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.5...v0.2.6
v0.2.6 is the most complete version of jaxDecomp Distributed FFTs now support jacrev, jacfwd and vmap
- Python
Published by ASKabalan about 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.5
What's Changed
- 60 remove shardedarray by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/61
- Updating readme and bumping version by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/62
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.4...v0.2.5
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.4
What's Changed
- 52 implement shardmapped ops by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/53
- update cuDecomp version by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/55
- PFFT returns ShardedArray if input is ShardedArray by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/56
- bump version to 0.2.4 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/57
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.3...v0.2.4
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.3
What's Changed
- bump jax version in dependencies by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/51
- Fix gradient issue with PFFT3D by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/49
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.2...v0.2.3
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.2
What's Changed
- bumping version by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/47
Releasing a new version with cleaned up build wheels that works for python 3.10 3.11 and 3.12
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.1...v0.2.2
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.1
What's Changed
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/32
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/33
- publish to pypi on release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/34
- remove condition for pypi release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/35
- Update github-deploy.yml by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/36
- 38 fix issue with cyclic import by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/39
- No testing on release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/40
- expand test matrix by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/41
- update python wheel api in toml by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/44
- Loosen JAX version constraint by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/45
- only build for x64 by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/46
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.0...v0.2.1
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.1
What's Changed
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/32
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/33
- publish to pypi on release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/34
- remove condition for pypi release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/35
- Update github-deploy.yml by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/36
- 38 fix issue with cyclic import by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/39
- No testing on release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/40
- expand test matrix by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/41
- As kabalan patch by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/42
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.0...v0.2.1
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.1
What's Changed
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/32
- Push to pypi the sequel by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/33
- publish to pypi on release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/34
- remove condition for pypi release by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/35
- Update github-deploy.yml by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/36
- 38 fix issue with cyclic import by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/39
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.2.0...v0.2.1
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.2.0
What's Changed
- One transpositions for slabs by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/27
- update install in README by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/29
- Implement jax versions by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/30
- Non contiguous transpose axis by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/28
- Docstrings by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/31
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.1.0...v0.2.0
- Python
Published by ASKabalan over 1 year ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - v0.1.0 first stable release of jaxDecomp
What's Changed
- Remove custom assignment of gpu by @EiffL in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/14
- Add a clang-format pre-commit and workflow by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/16
- Fix 3D FFT by @EiffL in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/12
- Implement Transpose primitives by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/17
- fix: some CMake fixes by @henryiii in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/21
- Fix slice unpad by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/18
- Clean up for JOSS paper by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/19
- [Packaging] Migrating to scikit-build core by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/22
- Proposal for simplified demo by @EiffL in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/24
- Joss paper of jaxDecomp by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/20
- Fix halo again by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/23
- update changelog by @ASKabalan in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/25
New Contributors
- @henryiii made their first contribution in https://github.com/DifferentiableUniverseInitiative/jaxDecomp/pull/21
Full Changelog: https://github.com/DifferentiableUniverseInitiative/jaxDecomp/compare/v0.0.1rc2...v0.1.0
- Python
Published by EiffL almost 2 years ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - Updated version compatible with jax v0.4.xx
- Python
Published by EiffL about 2 years ago
jaxDecomp: JAX Library for 3D Domain Decomposition and Parallel FFTs - First release candidate for initial API
Implements initial set of functionalities, in particular gives access to 3D FFT and halo exchange.
- Python
Published by EiffL over 3 years ago