Data

Tools

Indexes

Applications

Experiments

Open Sustainable Technology

https://github.com/mpi4jax/mpi4jax

mpi4jax: Zero-copy MPI communication of JAX arrays - Published in JOSS (2021)
gpu high-performance-computing jax jit mpi parallel-computing xla
Added: 10 months ago - Last Synced: 10 months ago - Created: July 21, 2020

  • Relevant topics? true
  • External users? true
  • Open source license? true
  • Active? true
  • Fork? false
  • Main Language: Python
  • Commits: 300
  • Committers: 7
  • Issues: 49
  • Pull Requests: 170
  • Owner: mpi4jax
  • Stars: 496
  • Forks: 33
  • Packages: 4
  • Downloads: 1,461