mpi4jax/mpi4jax issues and pull requests

#282 - Bump jax from 0.5.2 to 0.6.2 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] about 1 month ago
Labels: dependencies, python

#281 - Add python-version file to fix dependabot

Pull Request - State: closed - Opened by dionhaefner about 1 month ago

#280 - Examples for multi-node scaling study?

Issue - State: open - Opened by vsoch 3 months ago - 3 comments

#279 - Remove token mode

Pull Request - State: closed - Opened by dionhaefner 4 months ago - 2 comments

#278 - Fix default no token mode for older JAX versions

Pull Request - State: closed - Opened by macekmar 5 months ago - 3 comments

#277 - Bump jax from 0.5.1 to 0.5.2 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 5 months ago
Labels: dependencies, python

#276 - Bump jax from 0.5.0 to 0.5.1 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 5 months ago
Labels: dependencies

#274 - bug in 0.7

Issue - State: closed - Opened by PhilipVinc 6 months ago - 18 comments

#273 - Bump jax from 0.4.38 to 0.5.0 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 6 months ago - 4 comments
Labels: dependencies

#272 - Bump jax from 0.4.37 to 0.4.38 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 7 months ago
Labels: dependencies

#271 - Bump jax from 0.4.36 to 0.4.37 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 8 months ago
Labels: dependencies

#270 - Bump jax from 0.4.35 to 0.4.36 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 8 months ago
Labels: dependencies

#269 - Non blocking operations

Issue - State: open - Opened by NicoClinco 8 months ago - 1 comment

#267 - mpi4jax vs. mpi4py for cpu only

Issue - State: closed - Opened by kchenhyd 9 months ago - 2 comments

#266 - Fix xpu backend

Pull Request - State: closed - Opened by dionhaefner 9 months ago - 2 comments

#265 - Could mpi4jax be used to effectively interface C++?

Issue - State: open - Opened by Joshuaalbert 9 months ago - 7 comments

#264 - Bump jax from 0.4.33 to 0.4.34 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 10 months ago - 8 comments
Labels: dependencies

#263 - Jitting a function that takes `MPI.Intracomm` as argument

Issue - State: closed - Opened by mtagliazucchi 10 months ago - 6 comments

#261 - Ensure out_types is an interable on gpu

Pull Request - State: closed - Opened by dionhaefner 10 months ago

#260 - shallow_water example: TypeError: 'TokenType' object is not iterable

Issue - State: closed - Opened by benkirk 10 months ago - 10 comments

#259 - Fix docs

Pull Request - State: closed - Opened by dionhaefner 10 months ago - 1 comment

#258 - Update _latest_jax_version.txt to jax 0.4.33

Pull Request - State: closed - Opened by PhilipVinc 10 months ago

#257 - Rocm : second attempt

Pull Request - State: open - Opened by PhilipVinc 10 months ago - 3 comments

#256 - shallow_water example: TypeError: 'TokenType' object is not iterable

Issue - State: closed - Opened by benkirk 11 months ago - 3 comments

#255 - Bump jax from 0.4.31 to 0.4.32 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 11 months ago
Labels: dependencies

#254 - Debug setup-mpi action

Pull Request - State: closed - Opened by PhilipVinc 11 months ago

#253 - Fix for mpi4py>=4

Pull Request - State: closed - Opened by dionhaefner 12 months ago - 7 comments

#252 - fixthedocs

Pull Request - State: closed - Opened by dionhaefner 12 months ago

#251 - Bump jax from 0.4.30 to 0.4.31 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] 12 months ago - 2 comments
Labels: dependencies

#250 - maximum version mpi4py <4

Pull Request - State: closed - Opened by PhilipVinc 12 months ago

#249 - MPI4py 4 breaks our install

Issue - State: closed - Opened by PhilipVinc 12 months ago - 3 comments

#248 - jax 0.4.31 will break no token mode

Issue - State: closed - Opened by PhilipVinc about 1 year ago

#247 - Bump jax from 0.4.29 to 0.4.30 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] about 1 year ago
Labels: dependencies

#246 - Update install instructions to reflect the new jax cuda wheel policy

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago

#245 - Can't disable Intel OneAPI if present on machine

Issue - State: open - Opened by henryiii about 1 year ago - 3 comments

#244 - Autodetect if we have MPI_GPU support

Pull Request - State: open - Opened by PhilipVinc about 1 year ago

#243 - SendRecv: update to modern interface

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago - 2 comments

#242 - Louder warning if mpi4py/Cython not installed in non isolated builds

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago

#241 - Fixes for cuda pjrt plugin

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago - 7 comments

#240 - Improve build errors when --no-build-isolation

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago

#239 - Bump jax from 0.4.28 to 0.4.29 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] about 1 year ago
Labels: dependencies

#238 - Cleanup Sendrecv implementation

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago

#237 - Rename gpu -> cuda where relevant

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago - 2 comments

#236 - Compile with pip Nvidia packages...

Pull Request - State: closed - Opened by PhilipVinc about 1 year ago - 11 comments

#235 - Mpi4jax + jax[cuda12]

Issue - State: closed - Opened by PhilipVinc about 1 year ago - 7 comments

#234 - Bump jax from 0.4.27 to 0.4.28 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] about 1 year ago
Labels: dependencies

#233 - Bump jax from 0.4.26 to 0.4.27 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] about 1 year ago
Labels: dependencies