vdutor / SphericalHarmonicsLinks
Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
☆34Updated 2 months ago
Alternatives and similar repositories for SphericalHarmonics
Users that are interested in SphericalHarmonics are comparing it to the libraries listed below
Sorting:
- Matrix-free linear algebra in JAX.☆159Updated 2 months ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated this week
- S2FFT: Differentiable and accelerated spherical transforms☆215Updated last week
- Neural Emulator Architectures in JAX.☆23Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- Geometric kernels on manifolds, meshes and graphs☆286Updated last month
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Updated 6 months ago
- ☆14Updated 2 years ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Updated last year
- Normalizing Flows using JAX☆86Updated 2 years ago
- ☆52Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Updated last year
- Lightweight MCMC sampling for PyTorch Models aka My Corona Project☆52Updated 6 months ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆90Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated 2 weeks ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated this week
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 6 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆36Updated 2 years ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆33Updated last year
- A generic interface for linear algebra backends☆76Updated 2 months ago
- Stochastic Normalizing Flows☆79Updated 4 years ago
- Simulation-based inference in JAX☆45Updated 2 months ago
- ☆216Updated last week
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Stochastic Gradient MCMC for Jax☆18Updated 8 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆74Updated 5 years ago
- Library for normalizing flows and neural flows.☆26Updated 3 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆25Updated last year
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year