mpi4jax mpi4jax Q A · Discussions · GitHub

Hybrid MPI program with JAX
Q&A mtagliazucchi

Sort by: Latest activity

Q&A Discussions

Ask the community for help

You must be logged in to vote

Parallelizing certain portion of the code multiple times

YigitElma asked Feb 18, 2025 in Q&A · Closed · Unanswered

13
You must be logged in to vote

mpi4jax with non static argument?

elyurn asked Nov 11, 2024 in Q&A · Closed · Answered

1
You must be logged in to vote

Hybrid MPI program with JAX

mtagliazucchi asked Oct 2, 2024 in Q&A · Closed · Answered

5