jhueckelheim / PerforADLinks
Automatic Differentiation for high-performance stencil loops
☆13Updated 4 years ago
Alternatives and similar repositories for PerforAD
Users that are interested in PerforAD are comparing it to the libraries listed below
Sorting:
- Path Tracing in JAX☆73Updated 6 years ago
- Compiler with automatic differentiation☆49Updated 2 years ago
- ☆12Updated 5 years ago
- Supplementary materials for our SIGGRAPH 2022 paper☆28Updated 3 years ago
- Benchmarking various AD tools.☆110Updated last year
- Program Generator for Small-Scale Linear Algebra Applications☆31Updated 7 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- ☆108Updated last year
- ☆55Updated last year
- Python Binding to NVRTC☆79Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 5 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 5 years ago
- Scalable Probabilistic Programming Library☆21Updated 7 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago
- python package implementing a multivariate Horner scheme for efficiently evaluating multivariate polynomials☆31Updated 10 months ago
- cuASR: CUDA Algebra for Semirings☆44Updated 3 years ago
- EGGS, a method to speed up sparse matrix operations when the same sparsity is used for multiple times. This repo contains examples that s…☆26Updated 5 years ago
- ☆33Updated 5 years ago
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 6 years ago
- Linnea is an experimental tool for the automatic generation of optimized code for linear algebra problems.☆72Updated 5 months ago
- ☆24Updated 2 years ago
- Automatic Differentiation for Tensor Algebras☆28Updated 7 years ago
- Optimal approximate sampling from discrete probability distributions☆18Updated 4 years ago
- ☆16Updated 7 months ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆22Updated 3 years ago
- ☆80Updated 4 years ago
- a compiler for re-writing image processing functions in C++ to Halide☆24Updated 3 years ago
- Generic API for dispatch to Pyro backends.☆16Updated 3 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Updated 6 years ago