Iolo-Jones / ManifoldDiffusionGeometry
Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
☆41Updated 3 months ago
Alternatives and similar repositories for ManifoldDiffusionGeometry:
Users that are interested in ManifoldDiffusionGeometry are comparing it to the libraries listed below
- The code for "Diffusion Geometry" (2024).☆40Updated 9 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 6 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆23Updated last month
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆80Updated 8 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Matrix-free linear algebra in JAX.☆115Updated last month
- Exact OU processes with JAX☆43Updated 5 months ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆24Updated 7 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆27Updated 9 months ago
- Geometric kernels on manifolds, meshes and graphs☆240Updated last week
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated this week
- Differentiable and accelerated spherical transforms with JAX☆145Updated this week
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆59Updated last month
- Riemannian Optimization Using JAX☆48Updated last year
- Flow-matching algorithms in JAX☆85Updated 6 months ago
- ☆63Updated last year
- Code for Lie Symmetries SSL paper☆19Updated 11 months ago
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆11Updated 8 months ago
- ☆27Updated last month
- ☆151Updated 2 weeks ago
- Website for the book "The Elements of Differentiable Programming".☆13Updated 5 months ago
- Code for the paper Universal Physics Transformers☆94Updated 2 weeks ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆45Updated last week
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 9 months ago
- Representation Learning on Topological Domains☆79Updated this week
- Code for the book "The Elements of Differentiable Programming".☆73Updated 3 weeks ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year