Mpi4Jax | Anaconda.org

conda-forge / packages / mpi4jax 0.6.1

0

MPI support for JAX

copied from cf-staging / mpi4jax

License: MIT
Home: https://github.com/mpi4jax/mpi4jax
Development: https://github.com/mpi4jax/mpi4jax
Documentation: https://mpi4jax.readthedocs.io
742165 total downloads
Last upload: 29 days and 17 hours ago

Installers

linux-64 v0.6.1
osx-arm64 v0.6.1
osx-64 v0.6.1

conda install

To install this package run one of the following:
conda install conda-forge::mpi4jax

Description

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

© 2024 Anaconda, Inc. All Rights Reserved. (v4.0.6) Legal | Privacy Policy