google-research / spherical-cnnLinks
☆130Updated last week
Alternatives and similar repositories for spherical-cnn
Users that are interested in spherical-cnn are comparing it to the libraries listed below
Sorting:
- S2FFT: Differentiable and accelerated spherical transforms☆215Updated this week
- Running Jax in PyTorch Lightning☆119Updated last year
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Updated last year
- Multiple dispatch over abstract array types in JAX.☆136Updated last month
- ☆173Updated 2 years ago
- Stencil computations in JAX☆71Updated 2 months ago
- Run PyTorch in JAX. 🤝☆312Updated 3 months ago
- Diffusion models in PyTorch☆127Updated last week
- Matrix-free linear algebra in JAX.☆159Updated 2 months ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆25Updated 8 months ago
- Graph neural networks in JAX.☆68Updated last year
- Turn SymPy expressions into trainable JAX expressions.☆360Updated last week
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆94Updated this week
- Bayesian algorithm execution (BAX)☆55Updated 4 years ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Updated 2 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆214Updated 6 months ago
- Stainless neural networks in JAX☆34Updated last week
- Implementation of Forward Laplacian algorithm in JAX☆94Updated 2 months ago
- The ✨Magical✨ JAX ML Library.☆18Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated 3 weeks ago
- ☆10Updated 3 years ago
- A parallel ODE solver for PyTorch☆276Updated last year
- Automatic Functional Differentiation in JAX☆81Updated 4 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated this week
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆55Updated 4 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated last year
- Differentiable signal processing on the sphere for PyTorch☆634Updated this week
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆128Updated last year
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year