mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap: (by PhilipVinc)

Mpi4jax Alternatives

Similar projects and alternatives to mpi4jax

  • Dask

    32 mpi4jax VS Dask

    Parallel computing with task scheduling

  • horovod

    8 mpi4jax VS horovod

    Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

  • Scout Monitoring

    Free Django app performance insights with Scout Monitoring. Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.

    Scout Monitoring logo
  • extending-jax

    Extending JAX with custom C++ and CUDA code

  • Bulk

    A modern interface for implementing bulk-synchronous parallel programs.

  • devito

    DSL and compiler framework for automated finite-differences and stencil computation

  • einops

    19 mpi4jax VS einops

    Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

  • pyhpc-benchmarks

    A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python :rocket:

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo
NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a better mpi4jax alternative or higher similarity.

mpi4jax reviews and mentions

Posts with mentions or reviews of mpi4jax. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2021-02-03.
  • [D] Jax (or other libraries) when not using GPUs/TPUs but CPUs.
    2 projects | /r/MachineLearning | 3 Feb 2021
    I've seen a couple of posts of folks using JAX for scientific computing (e.g. physics) workloads without much issue. The parallel primitives work just as well across multiple CPUs as they do on accelerators. If you're on a cluster, also worth looking into https://github.com/PhilipVinc/mpi4jax.

Stats

Basic mpi4jax repo stats
1
380
6.7
23 days ago

PhilipVinc/mpi4jax is an open source project licensed under MIT License which is an OSI approved license.

The primary programming language of mpi4jax is Python.


Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com