google-research / spherical-cnn
☆114Updated 3 months ago
Alternatives and similar repositories for spherical-cnn:
Users that are interested in spherical-cnn are comparing it to the libraries listed below
- Differentiable and accelerated spherical transforms with JAX☆146Updated this week
- Run PyTorch in JAX. 🤝☆230Updated 3 weeks ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆43Updated 8 months ago
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago
- Stainless neural networks in JAX☆32Updated this week
- Running Jax in PyTorch Lightning☆89Updated 2 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆150Updated 3 months ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- Diffusion models in PyTorch☆96Updated this week
- Differentiable signal processing on the sphere for PyTorch☆440Updated 2 weeks ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆47Updated 2 weeks ago
- Stencil computations in JAX☆70Updated last year
- ☆17Updated 2 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 7 months ago
- Code for the book "The Elements of Differentiable Programming".☆75Updated last month
- Official implementation of Score-based Data Assimilation☆51Updated last year
- Fully and Partially Bayesian Neural Nets☆62Updated this week
- Pytorch-like dataloaders for JAX.☆76Updated 4 months ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆17Updated 10 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated last month
- Unleash the true power of scheduling☆28Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆330Updated last month
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- ☆53Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 5 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆23Updated last month
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated last week
- ☆151Updated last year
- A parallel ODE solver for PyTorch☆250Updated 5 months ago