pyhpc-benchmarks
mpi4jax
pyhpc-benchmarks | mpi4jax | |
---|---|---|
6 | 1 | |
301 | 371 | |
- | 3.2% | |
3.2 | 6.7 | |
4 months ago | 20 days ago | |
Python | Python | |
The Unlicense | MIT License |
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.
pyhpc-benchmarks
-
Supercharged high-resolution ocean simulation with Jax
True, but unfortunately Pytorch is not quite there yet when it comes to more complex benchmarks:
https://github.com/dionhaefner/pyhpc-benchmarks#example-resu...
JAX really is the only library that comes close to low-level code on CPU, almost always.
-
[D] Does working with Tensorflow affect my chances of getting research internships?
https://github.com/dionhaefner/pyhpc-benchmarks begs to differ.
- GitHub - dionhaefner/pyhpc-benchmarks: A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python
- HPC Benchmarks for Python
- Pyhpc: Benchmarks for CPU and GPU of the most popular high-perf Python libs
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?
tf-quant-finance - High-performance TensorFlow library for quantitative finance.
horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
pyopencl - OpenCL integration for Python, plus shiny features
extending-jax - Extending JAX with custom C++ and CUDA code
sqloxide - Python bindings for sqlparser-rs
Dask - Parallel computing with task scheduling
MATDaemon.jl
Bulk - A modern interface for implementing bulk-synchronous parallel programs.
3d-ken-burns - an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
devito - DSL and compiler framework for automated finite-differences and stencil computation
XLA.jl - "Maybe we have our own magic."
einops - Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)