vdutor / SphericalHarmonicsLinks
Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
☆34Updated last year
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.☆149Updated 2 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆116Updated 2 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆209Updated this week
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 8 months ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Updated 9 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆50Updated 4 months ago
- Geometric kernels on manifolds, meshes and graphs☆278Updated this week
- A generic interface for linear algebra backends☆75Updated this week
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆49Updated 4 months ago
- ☆198Updated 3 weeks ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Simulation-based inference in JAX☆40Updated 7 months ago
- Neural Emulator Architectures in JAX.☆20Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 5 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆74Updated 4 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆106Updated 2 weeks ago
- Stochastic Normalizing Flows☆78Updated 3 years ago
- Stainless neural networks in JAX☆34Updated 2 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Normalizing Flows using JAX☆85Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆209Updated 3 months ago
- Differentiable scientific computing library☆156Updated last year
- Gaussian Mixture Models in Jax☆23Updated 4 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆30Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆51Updated 3 weeks ago
- Newton and Quasi-Newton optimization with PyTorch☆367Updated 5 months ago
- Library for normalizing flows and neural flows.☆25Updated 3 years ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆252Updated 3 weeks ago