dionhaefner / pyhpc-benchmarks
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python
☆324Updated 6 months ago
Alternatives and similar repositories for pyhpc-benchmarks:
Users that are interested in pyhpc-benchmarks are comparing it to the libraries listed below
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆478Updated last month
- Extending JAX with custom C++ and CUDA code☆394Updated 8 months ago
- The Foundation for All Legate Libraries☆216Updated this week
- RFC document, tooling and other content related to the array API standard☆235Updated last month
- Mathematical operations for JAX pytrees☆199Updated 5 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆257Updated 2 years ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆75Updated 3 years ago
- Python bindings for UCX☆134Updated this week
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆901Updated last month
- Documentation:☆119Updated last year
- Utilities for Dask and CUDA interactions☆305Updated this week
- ☆848Updated this week
- Turn SymPy expressions into trainable JAX expressions.☆337Updated last week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆962Updated 3 weeks ago
- The CUDA target for Numba☆110Updated this week
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Provide Python access to the NVML library for GPU diagnostics☆234Updated 5 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆437Updated 2 weeks ago
- KvikIO - High Performance File IO☆206Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆391Updated this week
- Example Numba implementations of functions☆175Updated 2 years ago
- A modular system for machinable research code☆35Updated 3 weeks ago
- Machine Learning for HPC Workflows☆132Updated this week
- DeepHyper: Scalable Asynchronous Neural Architecture and Hyperparameter Search for Deep Neural Networks☆288Updated last week
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- Pyculib - Python bindings for CUDA libraries☆97Updated 6 years ago
- Automatic parallelization of Python/NumPy, C, and C++ codes on Linux and MacOSX☆220Updated 4 years ago
- PIX is an image processing library in JAX, for JAX.☆415Updated 2 months ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated 4 months ago