jcmgray / einsum_bmmLinks
einsum via batch matrix multiply
☆13Updated last year
Alternatives and similar repositories for einsum_bmm
Users that are interested in einsum_bmm are comparing it to the libraries listed below
Sorting:
- lazyarray is a Python package that provides a lazily-evaluated numerical array class, larray, based on and compatible with NumPy arrays.☆20Updated last year
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆38Updated last year
- Special function implementations☆20Updated last week
- Numerical derivatives for Python☆50Updated this week
- Use numba-compiled kernels from within Jax☆31Updated 5 months ago
- Mathematica package to convert Mathematica expressions to Numpy-compatible ones.☆13Updated last year
- Simple Xarray + JAX Integration☆21Updated last month
- Supplementary materials for the paper: Benchmarking the Parallel 1D Heat Equation Solver in Chapel, Charm++, C++, HPX, Go, Julia, Python,…☆14Updated last year
- SciPy fixes and extensions☆19Updated 3 years ago
- ☆35Updated last year
- Efficiently generate and analyse high dimensional data.☆71Updated last year
- Some useful extensions for NumPy☆29Updated last year
- A Python toolkit for coordinating asynchronous and dynamic ensembles of calculations.☆71Updated this week
- Slides/notes and Jupyter notebook demos for an introductory course of numerical methods for PDEs☆21Updated last year
- ☆22Updated 5 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆99Updated this week
- Numba @jit compatible wrappers for MPI C API tested on Linux, macOS and Windows☆44Updated this week
- Finite Element Solvers for Atomic Structure Calculations☆12Updated this week
- Explore HDF5 files in terminal & HTML views☆78Updated 11 months ago
- Krylov subspace methods package for Python☆107Updated 10 months ago
- Resampling tools for openPMD PIC data.☆24Updated last week
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆16Updated last year
- Python wrapper for the Cubature algorithm explained in: http://ab-initio.mit.edu/wiki/index.php/Cubature☆39Updated last month
- Strict implementation of the Python array API (previously numpy.array_api)☆27Updated last month
- Creates performance portable libraries with embedded source representations.☆26Updated 10 months ago
- A script that can help you find JOSS reviewers☆13Updated 5 years ago
- Evaluate layer and volume potentials accurately. Solve integral equations.☆27Updated this week
- Fortran Library for numerical INTegration of differential equations☆45Updated last year
- The Toolkit for Adaptive Stochastic Modeling and Non-Intrusive ApproximatioN☆72Updated last month
- Unit test demo using pFUnit.☆17Updated 8 years ago