lava
mpi4jax
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
lava
mpi4jax
-
[D] Jax (or other libraries) when not using GPUs/TPUs but CPUs.
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.
What are some alternatives?
magicavoxel-shaders - A collection of shaders for MagicaVoxel to generate geometry, noise, patterns, and simplify common and repetitive tasks.
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
python-performance - Repository for the book Fast Python - published by Manning
extending-jax - Extending JAX with custom C++ and CUDA code
Dask - Parallel computing with task scheduling
Bulk - A modern interface for implementing bulk-synchronous parallel programs.
devito - DSL and compiler framework for automated finite-differences and stencil computation
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: