DiffEqML / torchdyn

Related projects ⓘ

Alternatives and complementary repositories for torchdyn

• Zymrael / awesome-neural-ode
  A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…
  ☆1,346Updated 2 months ago
• msurtsukov / neural-ode
  Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations
  ☆692Updated 8 months ago
• google-research / torchsde
  Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
  ☆1,587Updated 5 months ago
• patrick-kidger / diffrax
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,445Updated this week
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆421Updated last year
• EmilienDupont / augmented-neural-odes
  Pytorch implementation of Augmented Neural ODEs
  ☆532Updated last year
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆629Updated 4 years ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆661Updated 2 years ago
• MilesCranmer / symbolic_deep_learning
  Code for "Discovering Symbolic Models from Deep Learning with Inductive Biases"
  ☆725Updated last year
• patrick-kidger / NeuralCDE
  Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
  ☆627Updated 2 years ago
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆933Updated 2 months ago
• jacobjinkelly / easy-neural-ode
  Code for the paper "Learning Differential Equations that are Easy to Solve"
  ☆270Updated 2 years ago
• rtqichen / torchdiffeq
  Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
  ☆5,597Updated last year
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆848Updated last year
• NeuroDiffGym / neurodiffeq
  A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, i…
  ☆702Updated 4 months ago
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆425Updated 2 months ago
• YuliaRubanova / latent_ode
  Code for "Latent ODEs for Irregularly-Sampled Time Series" paper
  ☆519Updated 3 years ago
• greydanus / hamiltonian-nn
  Code for our paper "Hamiltonian Neural Networks"
  ☆426Updated 3 years ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆471Updated this week
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆232Updated last month
• janosh / awesome-normalizing-flows
  Awesome resources on normalizing flows.
  ☆1,438Updated last month
• neuraloperator / neuraloperator
  Learning in infinite dimension with neural operators.
  ☆2,115Updated this week
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆526Updated this week
• boschresearch / torchphysics
  ☆409Updated last month
• cvxgrp / cvxpylayers
  Differentiable convex optimization layers
  ☆1,822Updated this week
• google / neural-tangents
  Fast and Easy Infinite Neural Networks in Python
  ☆2,279Updated 8 months ago
• titu1994 / tfdiffeq
  Tensorflow implementation of Ordinary Differential Equation Solvers with full GPU support
  ☆218Updated 4 years ago
• MilesCranmer / lagrangian_nns
  Lagrangian Neural Networks
  ☆465Updated 4 months ago
• google-deepmind / jraph
  A Graph Neural Network Library in Jax
  ☆1,376Updated 8 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,699Updated last week