eepperly / XTrace
Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"
☆12Updated last year
Alternatives and similar repositories for XTrace:
Users that are interested in XTrace are comparing it to the libraries listed below
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆9Updated 2 years ago
- Reverse-mode AD of dominant sparse eigensolver using Pytorch.☆38Updated 4 years ago
- Fast extremal eigensolvers for PyTorch.☆17Updated 3 years ago
- Julia package for tensors as multidimensional arrays, with functionalty within Tucker format, Kruskal (CP) format, Hierarchical Tucker fo…☆65Updated last year
- Orthogonal polynomials with JAX☆12Updated 2 weeks ago
- Code for Hutch++: Optimal Stochastic Trace Estimation☆11Updated 3 years ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Updated 5 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆53Updated last month
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- Differentiable interface to Firedrake for JAX☆14Updated 3 years ago
- Code for paper https://arxiv.org/abs/2306.07961☆53Updated 6 months ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- The Jacobi Davison diagonalization, python interface with fortran underlying.☆12Updated 5 years ago
- Easy interoperability with Automatic Differentiation libraries through NumPy interface to Firedrake and FEniCS☆14Updated last year
- Python Tensor Toolbox☆28Updated last week
- Efficient Differentiable n-d PDE solvers in JAX.☆22Updated 2 months ago
- A Julia implementation of tensor decomposition algorithms☆45Updated 4 years ago
- H2 Matrix Package☆25Updated last year
- A software package for flexible HPC GPs☆14Updated last month
- Cross interpolation of high-dimensional arrays in tensor train format☆22Updated 11 months ago
- A Python package for efficient optimisation of real-space renormalization group transformations using Tensorflow.☆28Updated 3 months ago
- A Julia library for hierarchical matrices☆44Updated 2 months ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆32Updated 2 years ago
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- Matrix-free linear algebra in JAX.☆113Updated last week
- WuLiXueBao magazine article - automatic differentiation.☆15Updated 2 years ago
- A Julia package for hierarchically semiseparable (HSS) matrices.☆25Updated last year
- ☆11Updated 4 months ago
- 18.A01: Seminar in Numerical Computing☆18Updated 6 years ago
- This project is used for the paper "Differentiable Programming of Isometric Tensor Network". (arXiv:2110.03898)☆18Updated 2 years ago