pnkraemer / matfreeLinks
Matrix-free linear algebra in JAX.
☆152Updated last week
Alternatives and similar repositories for matfree
Users that are interested in matfree are comparing it to the libraries listed below
Sorting:
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆117Updated last month
- Stencil computations in JAX☆71Updated 3 weeks ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Multiple dispatch over abstract array types in JAX.☆136Updated 2 weeks ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated 3 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated last year
- ☆205Updated last month
- A generic interface for linear algebra backends☆75Updated 3 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆100Updated 7 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.