No project description provided
Project description
- Jax-mpi provides integration among jax and MPI, so that
code containing MPI calls can be correctly jit-compiled through jax.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mpi4jax-0.2.7.tar.gz
(74.9 kB
view hashes)