cornellius-gp / linear_operator.old
A LinearOperator implementation for PyTorch
☆18Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for linear_operator.old
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Layered distributions using FLAX/JAX☆10Updated 3 years ago
- ☆12Updated 4 years ago
- Efficient Householder Transformation in PyTorch☆60Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated last month
- ☆78Updated 3 years ago
- Tree Approximate Message Passing☆29Updated 6 months ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆13Updated 9 months ago
- The differentiable OSQP solver layer for PyTorch☆59Updated last year
- Examples of more involved applications using Geomstats☆32Updated 3 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- Generic API for dispatch to Pyro backends.☆16Updated 2 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆58Updated last year
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago
- ☆15Updated 3 years ago
- Jax-based MaxEnt☆18Updated 4 years ago
- ☆22Updated 4 years ago
- AlgoPy is a Research Prototype for Algorithmic Differentation in Python☆80Updated 4 months ago
- ☆33Updated 3 years ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- ☆17Updated 2 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 4 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 6 years ago
- Hessian trace estimation using PyTorch and Hutch++☆18Updated 4 years ago