Project Awesome project awesome

New Libraries > mpi4jax

Combine MPI operations with your Jax code on CPUs and GPUs.

Package 520 stars GitHub
Back to JAX