RaphaelArkadyMeyerNYU / HutchPlusPlus
Code for Hutch++: Optimal Stochastic Trace Estimation
☆11Updated 3 years ago
Related projects: ⓘ
- Hierarchical variational models for physics.☆18Updated 4 years ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆11Updated last year
- Jax-based MaxEnt☆18Updated 4 years ago
- orbital MCMC☆10Updated 3 years ago
- ☆15Updated 3 years ago
- An example showing how to use jax to train resnet50 on multi-node multi-GPU☆20Updated 2 years ago
- A tool for weighted model counting through tensor network contraction☆20Updated 2 years ago
- A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.☆18Updated this week
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆34Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Practical tools for quantifying how well a sample approximates a target distribution☆27Updated 4 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- A Fast sketching based solver for large scale ridge regression☆17Updated 3 months ago
- Continuous-time gradient flow for generative modeling and variational inference☆29Updated 5 years ago
- A Python package for efficient optimisation of real-space renormalization group transformations using Tensorflow.☆27Updated 11 months ago
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆18Updated 3 months ago
- Riemannian Optimization Using JAX☆44Updated 10 months ago
- Forward mode laplacian implemented in JAX tracer☆27Updated 2 months ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆31Updated last year
- ☆22Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- ☆13Updated 3 years ago
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆17Updated 2 weeks ago
- Differentiable interface to Firedrake for JAX☆14Updated 3 years ago
- ☆78Updated 3 years ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆13Updated 7 months ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- ☆15Updated 3 years ago
- Code for Non-convex Learning via Replica Exchange Stochastic Gradient MCMC, ICML 2020.☆23Updated 3 years ago
- Relative gradient optimization of the Jacobian term in unsupervised deep learning, NeurIPS 2020☆21Updated 3 years ago