mpi4jax/mpi4jax issues and pull requests

#261 - Ensure out_types is an interable on gpu

Pull Request - State: closed - Opened by dionhaefner 6 days ago

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

Issue - State: closed - Opened by benkirk 8 days ago - 10 comments

#259 - Fix docs

Pull Request - State: closed - Opened by dionhaefner 11 days ago - 1 comment

#258 - Update _latest_jax_version.txt to jax 0.4.33

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

#257 - Rocm : second attempt

Pull Request - State: open - Opened by PhilipVinc 11 days ago

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

Issue - State: closed - Opened by benkirk 15 days 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] 16 days ago
Labels: dependencies

#254 - Debug setup-mpi action

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

#253 - Fix for mpi4py>=4

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

#252 - fixthedocs

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

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

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

#250 - maximum version mpi4py <4

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

#249 - MPI4py 4 breaks our install

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

#248 - jax 0.4.31 will break no token mode

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

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

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

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

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

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

Issue - State: open - Opened by henryiii 4 months ago - 3 comments

#244 - Autodetect if we have MPI_GPU support

Pull Request - State: open - Opened by PhilipVinc 4 months ago

#243 - SendRecv: update to modern interface

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

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

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

#241 - Fixes for cuda pjrt plugin

Pull Request - State: closed - Opened by PhilipVinc 4 months ago - 7 comments

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

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

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

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

#238 - Cleanup Sendrecv implementation

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

#237 - Rename gpu -> cuda where relevant

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

#236 - Compile with pip Nvidia packages...

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

#235 - Mpi4jax + jax[cuda12]

Issue - State: closed - Opened by PhilipVinc 4 months 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] 5 months 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] 5 months ago
Labels: dependencies

#232 - Missing AbstractToken attribute

Issue - State: closed - Opened by coreyjadams 5 months ago - 12 comments

#231 - Bump jax from 0.4.25 to 0.4.26 in /mpi4jax/_src

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

#230 - Fix #229

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

#229 - Unexpected behavior with JIT'ing allreduce

Issue - State: closed - Opened by coreyjadams 7 months ago - 10 comments

#228 - Bump jax from 0.4.24 to 0.4.25 in /mpi4jax/_src

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

#227 - Bump jax from 0.4.23 to 0.4.24 in /mpi4jax/_src

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

#226 - Extending mpi4jax with XPU support

Pull Request - State: closed - Opened by jczaja 8 months ago - 13 comments

#225 - Reduce operation enabled

Pull Request - State: closed - Opened by jczaja 8 months ago - 5 comments

#224 - Bump jax from 0.4.21 to 0.4.23 in /mpi4jax/_src

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

#223 - Bump jax from 0.4.20 to 0.4.21 in /mpi4jax/_src

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

#222 - bump flake8 version

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

#221 - Bump jax from 0.4.19 to 0.4.20 in /mpi4jax/_src

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

#220 - CUDA toolkit required if JAX with GPU support is installed via pip

Issue - State: closed - Opened by maxbortone 11 months ago - 8 comments

#219 - Bump jax from 0.4.18 to 0.4.19 in /mpi4jax/_src

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

#218 - Bump jax from 0.4.17 to 0.4.18 in /mpi4jax/_src

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

#217 - Bump jax from 0.4.16 to 0.4.17 in /mpi4jax/_src

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

#216 - Fix gpu translation rules that still use out_types keyword

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

#215 - unexpected keyword argument 'out_types'

Issue - State: closed - Opened by smartalecH 12 months ago - 2 comments

#214 - Drop python 3.7 and jax 0.3

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

#213 - Support for extended dtypes (new rng keys)

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

#212 - Experiment with ordered effects

Pull Request - State: closed - Opened by dionhaefner about 1 year ago - 24 comments

#211 - Incompatibility with jax 0.4.16

Issue - State: closed - Opened by inailuig about 1 year ago - 1 comment

#210 - Updates

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

#209 - Bump jax from 0.4.14 to 0.4.16 in /mpi4jax/_src

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

#208 - Is any model zoo or well-known case for mpi4jax deployment?

Issue - State: closed - Opened by Zantares about 1 year ago - 1 comment

#207 - Bump jax from 0.4.13 to 0.4.14 in /mpi4jax/_src

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

#206 - Can we support async API? like MPI_Iallreduce.

Issue - State: open - Opened by hbelove about 1 year ago - 1 comment

#205 - Bump jax from 0.4.12 to 0.4.13 in /mpi4jax/_src

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

#204 - Bump jax from 0.4.11 to 0.4.12 in /mpi4jax/_src

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

#203 - Fix auto tokeniser

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

#202 - Bump jax from 0.4.10 to 0.4.11 in /mpi4jax/_src

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

#201 - Bump jax from 0.4.9 to 0.4.10 in /mpi4jax/_src

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

#200 - Bump jax from 0.4.8 to 0.4.9 in /mpi4jax/_src

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

#199 - AMD's ROCM-HIP support

Pull Request - State: open - Opened by nutrik over 1 year ago - 11 comments

#198 - Bump jax from 0.4.7 to 0.4.8 in /mpi4jax/_src

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

#197 - Bump jax from 0.4.6 to 0.4.7 in /mpi4jax/_src

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

#196 - Bump jax from 0.4.5 to 0.4.6 in /mpi4jax/_src

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

#195 - Bump jax from 0.4.4 to 0.4.5 in /mpi4jax/_src

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

#194 - Two process running on the same GPU

Issue - State: closed - Opened by ntlm1686 over 1 year ago - 2 comments

#193 - [DoNotMerge] add test with custom_vjp that fails due to effects

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

#192 - auto_tokenize is broken for JAX >= 0.4.4

Issue - State: closed - Opened by dionhaefner over 1 year ago - 1 comment
Labels: bug, help wanted

#191 - Bump jax from 0.4.3 to 0.4.4 in /mpi4jax/_src

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

#190 - fixes for jax 0.4.4

Pull Request - State: closed - Opened by dionhaefner over 1 year ago - 3 comments

#189 - Investigate #188

Pull Request - State: closed - Opened by dionhaefner over 1 year ago - 2 comments

#188 - Error during compilation: No matching function to call, 1st argument ('const char *') would lose const qualifier

Issue - State: closed - Opened by louiskirsch over 1 year ago - 7 comments

#187 - Bump jax from 0.4.2 to 0.4.3 in /mpi4jax/_src

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

#186 - Bump jax from 0.4.1 to 0.4.2 in /mpi4jax/_src

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

#185 - Is collective communication within a specific group instead of the communicator supported?

Issue - State: closed - Opened by ntlm1686 over 1 year ago - 2 comments

#184 - How to manually run a distributed program without using mpirun?

Issue - State: closed - Opened by ntlm1686 almost 2 years ago - 1 comment

#183 - Bump jax from 0.4.0 to 0.4.1 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] almost 2 years ago - 1 comment
Labels: dependencies

#182 - Bump jax from 0.3.25 to 0.4.0 in /mpi4jax/_src

Pull Request - State: closed - Opened by dependabot[bot] almost 2 years ago - 4 comments
Labels: dependencies

#181 - ImportError: The mpi4jax GPU extensions could not be imported. Please re-build mpi4jax with CUDA support and try again.

Issue - State: closed - Opened by HuangLianghong almost 2 years ago - 3 comments

GitHub / mpi4jax/mpi4jax issues and pull requests