Abstract your array operations.
APACHE-2.0 License
High-Performance Klong array language in Python.
A collection of code snippets from the publication Daily Dose of Data Science on Substack: http:/...
N-D labeled arrays and datasets in Python
Stats, linear algebra and einops for xarray
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
Type and shape validation and serialization for numpy arrays in pydantic models
Python package for decomposing multi-dimensional arrays into sub-arrays (slices) according to mul...
Python library that implements named tensors with Astropy Quantity support.
TensorLy: Tensor Learning in Python.
Efficient argmin & argmax
C++ implementation of the Python Numpy library
Converters between Armadillo matrices (C++) and Numpy arrays using Pybind11
One tensor operation is all you need