vdutor / SphericalHarmonicsLinks
Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
☆34Updated last month
Alternatives and similar repositories for SphericalHarmonics
Users that are interested in SphericalHarmonics are comparing it to the libraries listed below
Sorting:
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆121Updated 2 weeks ago
- Matrix-free linear algebra in JAX.☆153Updated 3 weeks ago
- Tutorial materials of the Probabilistic Numerics Spring School.☆35Updated 2 years ago
- S2FFT: Differentiable and accelerated spherical transforms☆211Updated this week
- Code for Gaussian Score Matching Variational Inference☆35Updated 10 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆51Updated 5 months ago
- Neural Emulator Architectures in JAX.☆23Updated last year
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Updated 10 months ago
- A generic interface for linear algebra backends☆75Updated last month
- ☆52Updated 2 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆108Updated 2 weeks ago
- Code for efficiently sampling functions from GP(flow) posteriors☆74Updated 5 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆25Updated last year
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Stochastic Normalizing Flows☆78Updated 4 years ago
- Einstein Fields official implementation in JAX.☆67Updated 4 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆31Updated last year
- Geometric kernels on manifolds, meshes and graphs☆282Updated 2 weeks ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 6 months ago
- Implementation of Forward Laplacian algorithm in JAX☆92Updated last month
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆211Updated 4 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆36Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 3 months ago
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆39Updated 3 weeks ago
- Newton and Quasi-Newton optimization with PyTorch☆370Updated 7 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year
- Normalizing Flows using JAX☆85Updated 2 years ago
- Simulation-based inference in JAX☆42Updated last month