Mpi4jax Alternatives

Similar projects and alternatives to mpi4jax

Dask

32 11,982 9.7 Python mpi4jax VS Dask

Parallel computing with task scheduling
horovod

8 13,942 5.8 Python mpi4jax VS horovod

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

www.influxdata.com sponsored

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.
extending-jax

2 350 3.5 Python mpi4jax VS extending-jax

Extending JAX with custom C++ and CUDA code
Bulk

1 92 0.0 C++ mpi4jax VS Bulk

A modern interface for implementing bulk-synchronous parallel programs.
devito

2 518 9.8 Python mpi4jax VS devito

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

17 7,897 7.4 Python mpi4jax VS einops

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

6 300 3.2 Python mpi4jax VS pyhpc-benchmarks

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

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

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.

Suggest an alternative to mpi4jax

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.