jcmgray / einsum_bmm
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
- SciPy fixes and extensions☆18Updated 2 years ago
- Software for large and sparse SVD calculations☆19Updated this week
- A script that can help you find JOSS reviewers☆13Updated 4 years ago
- Minterpy - Multivariate interpolation in Python.☆35Updated 3 months ago
- CUDA training materials☆17Updated 4 months ago
- Use numba-compiled kernels from within Jax☆25Updated 4 months ago
- A conda-smithy repository for julia.☆23Updated 2 months ago
- Trajectory simulations for point particles in the Ocean & Atmosphere☆48Updated this week
- eigensolvers based on unitary core transformations☆22Updated 6 months ago
- Numba @jit compatible wrappers for MPI C API tested on Linux, macOS and Windows☆39Updated this week
- Repo to host book about Lagrangian Descriptors technique☆15Updated last year
- A lightweight, efficient, highly automated, fortran wrapper for python.☆72Updated 8 months ago
- Linnea is an experimental tool for the automatic generation of optimized code for linear algebra problems.☆68Updated 3 years ago
- ☆53Updated this week
- Set up your GitHub Actions workflow to use MPI☆31Updated last month
- ☆28Updated 3 years ago
- Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX☆23Updated last year
- Python combination of Ray and Numba providing compiled distributed arrays, remote functions, and actors.☆32Updated 8 months ago
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆33Updated last year
- Fortran Library for numerical INTegration of differential equations☆45Updated 11 months ago
- Collection of simple General Matrix Multiplication - GEMM implementations☆13Updated 10 months ago
- Symbolic code generators for multipole and local expansions and translations☆32Updated last week
- Parallel CPU and GPU high-performance computing - short course☆36Updated 3 years ago
- Efficiently generate and analyse high dimensional data.☆68Updated 4 months ago
- Python package for computing and visualizing Lagrangian Descriptors in Dynamical Systems☆20Updated 3 years ago
- Custom LaTeX2e documentclass for typesetting beautiful, modern theses.☆9Updated 2 years ago
- Some useful extensions for NumPy☆27Updated 6 months ago
- Stencil computations in JAX☆70Updated last year
- This repository mirrors the principal Gitlab repository of the Chebyshev Accelerated Subspace iteration Eigensolver. If you want to contr…☆16Updated 3 weeks ago
- Differentiable interface to FEniCS for PyMC3☆20Updated 2 years ago