imbirik / LieStationaryKernels

Related projects ⓘ

Alternatives and complementary repositories for LieStationaryKernels

• AaltoML / PeriodicBNN
  Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)
  ☆18Updated 3 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆16Updated 4 years ago
• wzhi / ODElearning_INN
  [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks
  ☆9Updated last year
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• aidanscannell / GPJax
  Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
  ☆11Updated 11 months ago
• glouppe / flowing-with-jax
  ☆15Updated 3 years ago
• alexOarga / haiku-geometric
  A collection of graph neural networks implementations in JAX
  ☆31Updated 11 months ago
• steindoringi / Variational_Integrator_Networks
  ☆18Updated last year
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated 2 years ago
• maxxxzdn / clifford-group-equivariant-cnns
  Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"
  ☆40Updated 3 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• ghliu / snopt
  Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight
  ☆52Updated 3 years ago
• pnkraemer / million-dimension-prob-ode-solver-experiments
  ☆10Updated 2 years ago
• aterenin / SparseGaussianProcesses.jl
  A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.
  ☆33Updated 4 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆13Updated 2 months ago
• aidos-lab / dect-evaluation
  Differentiable Euler Characteristic Transform
  ☆17Updated 5 months ago
• RussellTsuchida / snefy
  An implementation of squared neural families in PyTorch
  ☆11Updated 3 weeks ago
• facebookresearch / SSLForPDEs
  Code for Lie Symmetries SSL paper
  ☆19Updated 8 months ago
• google-deepmind / annealed_flow_transport
  ☆49Updated last year
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆25Updated last year