Zymrael / torchSODE

Related projects ⓘ

Alternatives and complementary repositories for torchSODE

• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆41Updated last year
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated 2 years ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 2 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• JSC370 / jsc370.github.io
  Course Website
  ☆9Updated 2 years ago
• necludov / iMCMC
  ☆22Updated 4 years ago
• steindoringi / Variational_Integrator_Networks
  ☆18Updated last year
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆46Updated 3 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• juntang-zhuang / torch_ACA
  repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE
  ☆55Updated 3 years ago
• mlysy / rodeo-legacy
  Probabilistic Solution of Differential Equations
  ☆14Updated 2 years ago
• afqueiruga / StatefulOdeNets
  Refining continuous-in-depth neural networks
  ☆39Updated 2 years ago
• clayton-r / symtorch
  Initial Commit
  ☆18Updated 3 years ago
• lorenzrichter / PDE-backward-solver
  ☆19Updated 2 weeks ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• duvenaud / jaxde
  Prototypes of differentiable differential equation solvers in JAX.
  ☆27Updated 4 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• nimadehmamy / L-conv-code
  Lie Algebra Convolutional Network implementation
  ☆42Updated 2 years ago
• juntang-zhuang / TorchDiffEqPack
  ☆45Updated last year
• kazukiosawa / ngd_in_wide_nn
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆14Updated 4 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago