https://github.com/nomonosound/numpy-minmax

A fast function (SIMD-accelerated) for finding the minimum and maximum value in a NumPy array
https://github.com/nomonosound/numpy-minmax

avx512 c float32 numpy numpy-library python simd x86-64

Last synced: about 1 month ago
JSON representation

A fast function (SIMD-accelerated) for finding the minimum and maximum value in a NumPy array

• Host: GitHub
• URL: https://github.com/nomonosound/numpy-minmax
• Owner: nomonosound
• License: mit
• Created: 2023-12-01T13:13:56.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-08-15T08:21:16.000Z (4 months ago)
• Last Synced: 2024-11-20T12:13:20.859Z (about 1 month ago)
• Topics: avx512, c, float32, numpy, numpy-library, python, simd, x86-64
• Language: Python
• Homepage:
• Size: 74.2 KB
• Stars: 11
• Watchers: 1
• Forks: 1
• Open Issues: 9
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# numpy-minmax: a fast function for finding the minimum and maximum value in a NumPy array

NumPy lacked an optimized minmax function, so we wrote our own. At Nomono, we use it for audio processing, but it can be applied any kind of float32 ndarray.

* Written in C and takes advantage of AVX/AVX512 for speed

* Roughly **2.3x speedup** compared to the numpy amin+amax equivalent (tested on Intel CPU with numpy 1.24-1.26)

* The fast implementation is tailored for float32 arrays that are C-contiguous, F-contiguous or 1D strided. Strided arrays with ndim >= 2 get processed with numpy.amin and numpy.amax, so no perf gain there. There is also a fast implementation for contiguous int16 arrays.

# Installation

[![PyPI version](https://img.shields.io/pypi/v/numpy-minmax.svg?style=flat)](https://pypi.org/project/numpy-minmax/)

![python 3.8, 3.9, 3.10, 3.11, 3.12](https://img.shields.io/badge/Python-3.8%20|%203.9%20|%203.10%20|%203.11|%203.12-blue)

![os: Linux, macOS, Windows](https://img.shields.io/badge/OS-Linux%20%28arm%20%26%20x86%29%20|%20macOS%20%28arm%29%20|%20Windows%20%28x86%29-blue)

```

$ pip install numpy-minmax

```

# Usage

```py

import numpy_minmax

import numpy as np

arr = np.arange(1337, dtype=np.float32)

min_val, max_val = numpy_minmax.minmax(arr)  # 0.0, 1336.0

```

# Changelog

## [0.3.1] - 2024-08-15

### Changes

* Optimize (with AVX) the processing of contiguous int16 arrays. ~2.3x speedup compared to 0.3.0

For the complete changelog, go to [CHANGELOG.md](CHANGELOG.md)

# Development

* Install dev/build/test dependencies as denoted in pyproject.toml

* `CC=clang pip install -e .`

* `pytest`

# Running benchmarks

* Install diplib `pip install diplib`

* `python scripts/perf_benchmark.py`

# Acknowledgements

This library is maintained/backed by [Nomono](https://nomono.co/), a Norwegian audio AI startup.