A parallel ODE solver for PyTorch
☆277Oct 3, 2024Updated last year
Alternatives and similar repositories for torchode
Users that are interested in torchode are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69May 22, 2024Updated last year
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,939Feb 23, 2026Updated last month
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,554May 2, 2024Updated last year
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,371Apr 4, 2025Updated 11 months ago
- TorchCFM: a Conditional Flow Matching library☆2,357Nov 11, 2025Updated 4 months ago
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,708Dec 30, 2024Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…☆1,521Sep 13, 2024Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- This is the reference implementation of our NeurIPS 2023 paper "Add and Thin: Diffusion for Temporal Point Processes"☆22Mar 4, 2024Updated 2 years ago
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 4 years ago
- Convolutions and more as einsum for PyTorch☆17Jun 6, 2024Updated last year
- code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".☆664Sep 22, 2020Updated 5 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,821Mar 9, 2026Updated 2 weeks ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆161Feb 20, 2023Updated 3 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Mar 4, 2026Updated 2 weeks ago
- ☆50Oct 28, 2023Updated 2 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆82Apr 19, 2025Updated 11 months ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆311Mar 5, 2026Updated 2 weeks ago
- Experiments for Neural Flows paper☆102Nov 24, 2021Updated 4 years ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- Turning SymPy expressions into PyTorch modules.☆158Jul 24, 2023Updated 2 years ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆91Feb 14, 2025Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated last month
- Pytorch implementation of Augmented Neural ODEs☆555Feb 9, 2023Updated 3 years ago
- A PyTorch library for all things nonlinear control and reinforcement learning.☆48Oct 22, 2021Updated 4 years ago
- ☆13Oct 24, 2021Updated 4 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Dec 10, 2022Updated 3 years ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆58Nov 16, 2023Updated 2 years ago
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 2 months ago
- ☆35Nov 22, 2024Updated last year
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆476Sep 4, 2025Updated 6 months ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆76May 1, 2022Updated 3 years ago
- A differentiable PDE solving framework for machine learning☆1,838Updated this week
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Nov 20, 2021Updated 4 years ago
- Voicebox: Text-Guided Multilingual Universal Speech Generation at Scale☆28Aug 4, 2023Updated 2 years ago
- Integrating temporal gene expression modalities for trajectory inference and disease prediction☆10Sep 20, 2022Updated 3 years ago
- A minimalist implementation of score-based diffusion model☆128Aug 18, 2021Updated 4 years ago
- Adaptive Vocoder for Custom Voice☆61Sep 22, 2022Updated 3 years ago