Iolo-Jones / ManifoldDiffusionGeometry

Related projects ⓘ

Alternatives and complementary repositories for ManifoldDiffusionGeometry

• Iolo-Jones / DiffusionGeometry
  The code for "Diffusion Geometry" (2024).
  ☆33Updated 5 months ago
• maxxxzdn / clifford-group-equivariant-cnns
  Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"
  ☆40Updated 3 months ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆41Updated last year
• ml-jku / GINNs-Geometry-informed-Neural-Networks
  ☆89Updated this week
• normal-computing / thermox
  Exact OU processes with JAX
  ☆34Updated last month
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆63Updated 2 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆103Updated 2 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆74Updated 3 months ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆76Updated 5 months ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• david-knigge / enf-min-jax
  A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
  ☆15Updated 2 months ago
• qiauil / ConvDO
  Convolutional Differential Operators for Physics-based Deep Learning Study
  ☆23Updated 3 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆38Updated 2 weeks ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 4 months ago
• alexandersoen / pyBregMan
  ☆25Updated 2 weeks ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated last year
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆32Updated 11 months ago
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆34Updated 2 weeks ago
• geometric-kernels / GeometricKernels
  Geometric kernels on manifolds, meshes and graphs
  ☆218Updated last week
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆29Updated last year
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆47Updated last year
• maxxxzdn / implicit-steerable-kernels
  Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.
  ☆26Updated 7 months ago
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆52Updated 2 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆102Updated this week
• plainerman / Variational-Doob
  Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling
  ☆37Updated 2 weeks ago
• pyt-team / TopoEmbedX
  Representation Learning on Topological Domains
  ☆78Updated 2 weeks ago
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆69Updated last year
• wesselb / lab
  A generic interface for linear algebra backends
  ☆70Updated 4 months ago