jhueckelheim / PerforAD
Automatic Differentiation for high-performance stencil loops
☆12Updated 4 years ago
Alternatives and similar repositories for PerforAD:
Users that are interested in PerforAD are comparing it to the libraries listed below
- Compiler with automatic differentiation☆48Updated last year
- Code accompanying the paper "Automating Involutive MCMC using Probabilistic and Differentiable Programming"☆13Updated 4 years ago
- Path Tracing in JAX☆71Updated 5 years ago
- Program Generator for Small-Scale Linear Algebra Applications☆29Updated 6 years ago
- tokenizer and parser for circle projects☆11Updated 5 years ago
- ☆33Updated 4 years ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- Supplementary materials for our SIGGRAPH 2022 paper☆28Updated 3 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆12Updated 4 years ago
- ☆51Updated 9 months ago
- ☆103Updated 10 months ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- JAX for Graphcore IPU (experimental)☆21Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Generic API for dispatch to Pyro backends.☆16Updated 3 years ago
- Scalable Probabilistic Programming Library☆21Updated 6 years ago
- Examples of Automatic Differentiation (AD) in many different languages and systems☆26Updated 6 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Optimal approximate sampling from discrete probability distributions☆17Updated 4 years ago
- Python Binding to NVRTC☆79Updated 7 months ago
- a fully-differentiable graphical raytracer☆15Updated 9 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- ☆15Updated 4 years ago
- ☆22Updated 2 years ago
- cuASR: CUDA Algebra for Semirings☆35Updated 2 years ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago