Alternatives and similar repositories for torchsde

Users that are interested in torchsde are comparing it to the libraries listed below

• DiffEqML / torchdyn
  A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
  ☆1,521Updated last year
• Zymrael / awesome-neural-ode
  A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…
  ☆1,476Updated last year
• patrick-kidger / diffrax
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,777Updated 2 weeks ago
• google / neural-tangents
  Fast and Easy Infinite Neural Networks in Python
  ☆2,359Updated last year
• msurtsukov / neural-ode
  Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations
  ☆779Updated last year
• patrick-kidger / NeuralCDE
  Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
  ☆666Updated 2 years ago
• patrick-kidger / torchcde
  Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
  ☆452Updated last month
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆959Updated 9 months ago
• cvxpy / cvxpylayers
  Differentiable convex optimization layers
  ☆1,985Updated last week
• AdamCobb / hamiltorch
  PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
  ☆458Updated last year
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,532Updated last week
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆997Updated 5 months ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆517Updated 5 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆2,027Updated this week
• EmilienDupont / augmented-neural-odes
  Pytorch implementation of Augmented Neural ODEs
  ☆552Updated 2 years ago
• janosh / awesome-normalizing-flows
  Awesome resources on normalizing flows.
  ☆1,568Updated 2 months ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆698Updated 2 months ago
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆657Updated 5 years ago
• google / edward2
  A simple probabilistic programming language.
  ☆699Updated 3 weeks ago
• pyro-ppl / numpyro
  Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
  ☆2,535Updated this week
• cornellius-gp / gpytorch
  A highly efficient implementation of Gaussian Processes in PyTorch
  ☆3,773Updated last month
• MilesCranmer / symbolic_deep_learning
  Code for "Discovering Symbolic Models from Deep Learning with Inductive Biases"
  ☆765Updated last year
• YuliaRubanova / latent_ode
  Code for "Latent ODEs for Irregularly-Sampled Time Series" paper
  ☆562Updated 4 years ago
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,924Updated last month
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,102Updated this week
• karpathy / pytorch-normalizing-flows
  Normalizing flows in PyTorch. Current intended use is education not production.
  ☆881Updated 5 years ago
• VincentStimper / normalizing-flows
  PyTorch implementation of normalizing flow models
  ☆887Updated last year
• jacobjinkelly / easy-neural-ode
  Code for the paper "Learning Differential Equations that are Easy to Solve"
  ☆283Updated 3 years ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.
  ☆644Updated this week
• piEsposito / blitz-bayesian-deep-learning
  A simple and extensible library to create Bayesian Neural Network layers on PyTorch.
  ☆973Updated 2 years ago