vdutor / SphericalHarmonics
Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
☆24Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for SphericalHarmonics
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆66Updated 4 years ago
- A generic interface for linear algebra backends☆70Updated 4 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆34Updated last year
- ☆49Updated last year
- ☆15Updated 2 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆51Updated 3 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆51Updated 8 months ago
- Riemannian Optimization Using JAX☆45Updated last year
- JAX exponential map normalising flows on sphere☆17Updated 4 years ago
- Implementation of Action Matching☆36Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆21Updated 6 months ago
- Code for Gaussian Score Matching Variational Inference☆28Updated last month
- Stochastic Normalizing Flows☆73Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Bayesian algorithm execution (BAX)☆48Updated 3 years ago
- Squared Non-monotonic Probabilistic Circuits☆19Updated 4 months ago
- Riemannian Convex Potential Maps☆68Updated last year
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated 11 months ago
- Stainless neural networks in JAX☆31Updated last month
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 5 months ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆29Updated 2 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆34Updated 3 weeks ago
- ☆107Updated 3 weeks ago