Data

Tools

Indexes

Applications

Experiments

Open Source Science

Recent Releases of mpi4jax

mpi4jax - v0.8.1

What's Changed

  • Bump jax from 0.6.2 to 0.7.0 in /mpi4jax/_src by @dependabot[bot] in https://github.com/mpi4jax/mpi4jax/pull/283

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.8.0...v0.8.1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner 10 months ago

mpi4jax - v0.8.0

⚠️ This is an API breaking release ⚠️

Token handling is now removed from mpi4jax. That is, communication primitives do not accept / return token objects anymore, and order of execution is enforced internally by JAX.

To upgrade, please adjust your code like this:

```python

BEFORE (mpi4jax<0.8.0)

result, token = mpi4jax.allgather(x, token=token)

AFTER (mpi4jax>=0.8.0)

result = mpi4jax.allgather(x) ```

What's Changed

  • Remove token mode by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/279

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.7.2...v0.8.0

Scientific Software - Peer-reviewed - Python
Published by dionhaefner 11 months ago

mpi4jax - v0.7.2

This is the final release that supports passing / returning tokens explicitly.

What's Changed

  • Add python-version file to fix dependabot by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/281
  • Bump jax from 0.5.2 to 0.6.2 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/282

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.7.1...v0.7.2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner 11 months ago

mpi4jax - v0.7.1

This is the last release that supports token mode and JAX versions < 0.5.1.

What's Changed

  • Fix default no token mode for older JAX versions by @macekmar in https://github.com/mpi4jax/mpi4jax/pull/278

New Contributors

  • @macekmar made their first contribution in https://github.com/mpi4jax/mpi4jax/pull/278

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.7.0...v0.7.1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 1 year ago

mpi4jax - v0.7.0.post1

What's Changed

  • Bump jax from 0.5.0 to 0.5.1 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/276
  • Bump jax from 0.5.1 to 0.5.2 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/277

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.7.0...v0.7.0.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 1 year ago

mpi4jax - v0.7.0

What's Changed

  • Drop support for Python<3.10.
  • Drop support for JAX<0.4.35.
  • New public module mpi4jax.notoken. Removed mpi4jax.experimental.notoken.
  • All primitives now use "notoken" mode under the hood (can be disabled via MPI4JAX_PREFER_NOTOKEN=0).

Please report any issues with these changes.

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.6.1...v0.7.0

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.6.1.post3

What's Changed

  • Bump jax from 0.4.37 to 0.4.38 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/272

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.6.1.post2...v0.6.1.post3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.6.1.post2

What's Changed

  • Bump jax from 0.4.36 to 0.4.37 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/271

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.6.1.post1...v0.6.1.post2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.6.1.post1

What's Changed

  • Bump jax from 0.4.35 to 0.4.36 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/270

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.6.1...v0.6.1.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.6.1

What's Changed

  • Fix xpu backend by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/266

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.6.0...v0.6.1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.6.0

What's Changed

  • Bump jax from 0.4.33 to 0.4.35 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/264
  • Removed support for intel XPU (for the moment)
  • Switch to jax FFI interface

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.4...v0.6.0

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 1 year ago

mpi4jax - v0.5.4

What's Changed

  • Bump jax from 0.4.31 to 0.4.32 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/255
  • Update latestjax_version.txt to jax 0.4.33 by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/258
  • Fix docs by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/259
  • Ensure out_types is an interable on gpu by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/261

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3...v0.5.4

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 1 year ago

mpi4jax - v0.5.3.post2

What's Changed

  • Update latestjax_version.txt to jax 0.4.33 by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/258
  • Fix docs by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/259

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3.post1...v0.5.3.post2

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 1 year ago

mpi4jax - v0.5.3.post1

What's Changed

  • Bump jax from 0.4.31 to 0.4.32 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/255

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.3...v0.5.3.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 1 year ago

mpi4jax - v0.5.3 : Support for MPI4py 4

What's Changed

  • Fix for mpi4py>=4 by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/253

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.2...v0.5.3

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 2 years ago

mpi4jax - v0.5.2

What's Changed

  • maximum version mpi4py <4 by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/250
  • Bump jax from 0.4.30 to 0.4.31 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/251
  • fixthedocs by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/252

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.1...v0.5.2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 2 years ago

mpi4jax - v0.5.1.post1

What's Changed

  • maximum version mpi4py <4 by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/250

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.1...v0.5.1.post1

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 2 years ago

mpi4jax - v0.5.1

What's Changed

  • Rename gpu -> cuda where relevant by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/237
  • Bump jax from 0.4.28 to 0.4.29 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/239
  • Improve build errors when --no-build-isolation by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/240
  • Cleanup Sendrecv implementation by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/238
  • Louder warning if mpi4py/Cython not installed in non isolated builds by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/242
  • Fixes for cuda pjrt plugin by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/241
  • Update install instructions to reflect the new jax cuda wheel policy by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/246
  • Bump jax from 0.4.29 to 0.4.30 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/247

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.0...v0.5.1

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 2 years ago

mpi4jax - v0.5.0.post1

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.5.0...v0.5.0.post1

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 2 years ago

mpi4jax - v0.5.0

What's Changed

mpi4jax does not require the CUDA toolkit to be installed anymore. Instead, if the pypi Nvidia wheels are installed already (which is the new default behaviour of jax) we can link to them directly.

  • Compile with pip Nvidia packages... by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/236

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.1...v0.5

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 2 years ago

mpi4jax - v0.4.1.post1

What's Changed

  • Bump jax from 0.4.26 to 0.4.27 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/233
  • Bump jax from 0.4.27 to 0.4.28 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/234

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.1...v0.4.1.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 2 years ago

mpi4jax - v0.4.1

What's Changed

  • Fix #229 by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/230
  • Extending mpi4jax with XPU support by @jczaja in https://github.com/mpi4jax/mpi4jax/pull/226

New Contributors

  • @jczaja made their first contribution in https://github.com/mpi4jax/mpi4jax/pull/226

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post6...v0.4.1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 2 years ago

mpi4jax - v0.4.0.post6

What's Changed

  • Bump jax from 0.4.25 to 0.4.26 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/231

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post5...v0.4.0.post6

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 2 years ago

mpi4jax - v0.4.0.post5

What's Changed

  • Bump jax from 0.4.24 to 0.4.25 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/228

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post4...v0.4.0.post5

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.4.0.post4

What's Changed

  • Bump jax from 0.4.23 to 0.4.24 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/227

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post3...v0.4.0.post4

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.4.0.post3

What's Changed

  • Bump jax from 0.4.21 to 0.4.23 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/224

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post2...v0.4.0.post3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.4.0.post2

What's Changed

  • Bump jax from 0.4.20 to 0.4.21 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/223

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0.post1...v0.4.0.post2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.4.0.post1

What's Changed

  • bump flake8 version by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/222
  • Bump jax from 0.4.19 to 0.4.20 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/221

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.4.0...v0.4.0.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.4.0

What's Changed

This release adds experimental support for mpi4jax primitives that do not require manual token management, located in mpi4jax.experimental.notoken. To test the new notoken mode in existing mpi4jax code, use MPI4JAX_PREFER_NOTOKEN=1.

  • mpi4jax.experimental.notoken module added by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/212
  • mpi4jax.experimental.auto_tokenize is removed.

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.15...0.4.0

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 2 years ago

mpi4jax - v0.3.15.post2

What's Changed

  • Bump jax from 0.4.17 to 0.4.18 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/218

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.15.post1...v0.3.15.post2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.3.15.post1

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 2 years ago

mpi4jax - v0.3.15

What's Changed

  • Fix gpu translation rules that still use out_types keyword by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/216

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14...v0.3.15

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.3.14.post9

What's Changed

  • Updates by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/210
  • Bump jax from 0.4.14 to 0.4.16 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/209

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post8...v0.3.14.post9

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 2 years ago

mpi4jax - v0.3.14.post8

What's Changed

  • Bump jax from 0.4.13 to 0.4.14 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/207

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post7...v0.3.14.post8

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 3 years ago

mpi4jax - v0.3.14.post7

What's Changed

  • Bump jax from 0.4.12 to 0.4.13 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/205

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post6...v0.3.14.post7

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 3 years ago

mpi4jax - v0.3.14.post6

What's Changed

  • Bump jax from 0.4.11 to 0.4.12 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/204

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post5...v0.3.14.post6

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 3 years ago

mpi4jax - v0.3.14.post5

What's Changed

  • Bump jax from 0.4.10 to 0.4.11 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/202

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post4...v0.3.14.post5

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 3 years ago

mpi4jax - v0.3.14.post4

What's Changed

  • Bump jax from 0.4.8 to 0.4.9 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/200
  • Bump jax from 0.4.9 to 0.4.10 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/201

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post3...v0.3.14.post4

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 3 years ago

mpi4jax - v0.3.13.post3

What's Changed

  • Bump jax from 0.4.7 to 0.4.8 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/198

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post2...v0.3.14.post3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 3 years ago

mpi4jax - v0.3.14.post2

What's Changed

  • Bump jax from 0.4.6 to 0.4.7 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/197

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14.post1...v0.3.14.post2

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 3 years ago

mpi4jax - v0.3.14.post1

What's Changed

  • Bump jax from 0.4.5 to 0.4.6 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/196

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.14...v0.3.14.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 3 years ago

mpi4jax - v0.3.14

What's Changed

  • Bump jax from 0.4.4 to 0.4.5 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/195

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.13...v0.3.14

Scientific Software - Peer-reviewed - Python
Published by dionhaefner about 3 years ago

mpi4jax - v0.3.13

What's Changed

  • allow MPIEffect in jax.custom_vjp. by @PhilipVinc in https://github.com/mpi4jax/mpi4jax/pull/166

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.12...v0.3.13

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 3 years ago

mpi4jax - v0.3.12

What's Changed

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.11...v0.3.12

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.11.post3

What's Changed

  • Bump jax from 0.4.2 to 0.4.3 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/187

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.11.post2...v0.3.11.post3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.11.post2 : support jax 0.4.2

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 3 years ago

mpi4jax - v0.3.11.post1

What's Changed

  • Bump jax from 0.3.25 to 0.4.0 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/182
  • Bump jax from 0.4.0 to 0.4.1 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/183

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.11...v0.3.11.post1

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.11

What's Changed

  • Port from XLA CustomCall to MLIR by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/179
  • Fixed a bug that led to incorrect results whenever XLA eliminated consecutive transposes (#176).
  • Now requires JAX >= 0.3.25.

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10...v0.3.11

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post7

What's Changed

  • Bump jax from 0.3.23 to 0.3.24 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/177

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10.post6...v0.3.10.post7

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post6

What's Changed

  • Bump jax from 0.3.22 to 0.3.23 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/173

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10.post5...v0.3.10.post6

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post5

What's Changed

  • Bump jax from 0.3.21 to 0.3.22 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/172

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10.post4...v0.3.10.post5

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post4

What's Changed

  • Bump jax from 0.3.20 to 0.3.21 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/171

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10.post3...v0.3.10.post4

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post3

What's Changed

  • Bump jax from 0.3.19 to 0.3.20 in /mpi4jax/_src by @dependabot in https://github.com/mpi4jax/mpi4jax/pull/170

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.10.post2...v0.3.10.post3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 3 years ago

mpi4jax - v0.3.10.post2 : bump jax

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 3 years ago

mpi4jax - Bump jax version for warning

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 3 years ago

mpi4jax - v0.3.10

What's Changed

  • Fix editable installs by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/164
  • Bump jax from 0.3.15 to 0.3.16
  • Add envvar to configure mpicc by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/165

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.9...v0.3.10

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 4 years ago

mpi4jax - v0.3.9

What's Changed

  • Use mpicc to build extensions by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/157
  • Support Jax 0.3.15 (changes to primitive's side-effect API) by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/151
  • Deprecate JAX 0.2.x by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/161
  • Fix docs on RTD by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/162

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.8...v0.3.9

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 4 years ago

mpi4jax - v0.3.8: fix bug related to re-entrant GPU callbacks

What's Changed

  • Do not use thread-local state (fixes #139) by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/140

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.7...v0.3.8

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 4 years ago

mpi4jax - v0.3.7

What's Changed

  • Add missing barrier primitive registration on GPU by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/118
  • Created auto_tokenize by @Thenerdstation in https://github.com/mpi4jax/mpi4jax/pull/111

New Contributors

  • @Thenerdstation made their first contribution in https://github.com/mpi4jax/mpi4jax/pull/111

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.6...v0.3.7

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 4 years ago

mpi4jax - v0.3.6

What's Changed

  • Restore support for Mac ARM
  • Only retrieve dtype object when needed (fixes #112) by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/113

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.5...v0.3.6

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 4 years ago

mpi4jax - v0.3.5

What's Changed

  • Fix inconsistent output shapes by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/110
  • Now supports jax >= 0.2.27.

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.4...v0.3.5

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 4 years ago

mpi4jax - v0.3.4

What's Changed

  • Adds mpi4jax.has_cuda_support to probe if mpi4jax was built with CUDA support

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.3...v0.3.4

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 4 years ago

mpi4jax - v0.3.3

What's Changed

  • Compatibility with jax>=0.2.26 by @dionhaefner in https://github.com/mpi4jax/mpi4jax/pull/108

Full Changelog: https://github.com/mpi4jax/mpi4jax/compare/v0.3.2...v0.3.3

Scientific Software - Peer-reviewed - Python
Published by dionhaefner over 4 years ago

mpi4jax - v0.3.2

  • Support for installing development dependencies via extras: pip install mpi4jax[dev]
  • Better debug logging and bug fixes

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 5 years ago

mpi4jax - v0.3.1

Compatibility with Jax 0.2.14

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 5 years ago

mpi4jax - v0.3.0

API changes

  • Most optional parameters of primitives (e.g. comm or token) are now keyword-only
  • Removed deprecated upper-cases aliases (e.g. use send instead of Send)

New features

  • New primitive barrier
  • Printing of MPI error strings on error
  • sendrecv is now differentiable

Bug fixes

  • Primitive outputs are now always tuples, not lists
  • Debug messages are flushed immediately

Scientific Software - Peer-reviewed - Python
Published by dionhaefner almost 5 years ago

mpi4jax - v0.2.18

Add support for jax.vmap to mpi4jax.allreduce

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.17

Bug fix for vjp of allreduce

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.16

remove setuptools runtime dependency

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.15

fix more sdist issues

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.14

Fix conda versioning issues by switching to versioneer.

No functional changes.

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.13

Fix tests

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.12

Add several new communication primitives.

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.11

  • Deprecate function names and change them to be uppercase
  • Bump minimum jaxlib version
  • Flush GPU queue on exit
  • Fix versioning issues on python 3.6

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc about 5 years ago

mpi4jax - v0.2.10

GPU Compilation fix

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.9: Support more types

Add support for np.float128, np.uint and such

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.8: Bcast support

Support for MPI Broadcast

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.7: GPU support and Allreduce Transposition

  • Experimental GPU Support #29 (@dionhaefner)
  • Linear Transposition of MPI.Allreduce #35 (@inailuig)

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.6

  • Automatically flush the CPU execution cache atexit
  • Improve handling of AD through AllReduce
  • Always use Cython implementations of MPI operations
  • Enforce type checking

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.5

Supports AD for allreduce.

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.3

Improving the build system yet again...

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.2

Fixes to published version

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.1

  • Raise errors in jitted code
  • Dont require Cython to be already installed when installed as a dependency

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - v0.2.0

Adds Send/Recv/SendRecv operations.

  • A token system has been introduced, to force data dependency among different MPI operations to prevent deadlock.

Closed Issues: - Operations now support numpy arrays even when not jitted (#5) - Operations now support scalars (#4)

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc over 5 years ago

mpi4jax - initial release

Scientific Software - Peer-reviewed - Python
Published by PhilipVinc almost 6 years ago