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.1.tar.gz
(4.2 kB
view hashes)