A parallel ODE solver for PyTorch
☆276Oct 3, 2024Updated last year
Alternatives and similar repositories for torchode
Users that are interested in torchode are comparing it to the libraries listed below
Sorting:
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,912Feb 23, 2026Updated last week
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,553May 2, 2024Updated last year
- TorchCFM: a Conditional Flow Matching library☆2,319Nov 11, 2025Updated 3 months ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69May 22, 2024Updated last year
- A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, i…☆771Jul 27, 2025Updated 7 months ago
- Differentiable SDE solvers with GPU support and efficient sensitivity analysis.☆1,704Dec 30, 2024Updated last year
- A collection of resources regarding the interplay between differential equations, deep learning, dynamical systems, control and numerical…☆1,517Sep 13, 2024Updated last year
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆161Feb 20, 2023Updated 3 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".☆664Sep 22, 2020Updated 5 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆122Updated this week
- Turning SymPy expressions into PyTorch modules.☆157Jul 24, 2023Updated 2 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆82Apr 19, 2025Updated 10 months ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Jan 12, 2026Updated last month
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆58Nov 16, 2023Updated 2 years ago
- ☆50Oct 28, 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 minimalist implementation of score-based diffusion model☆128Aug 18, 2021Updated 4 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆20Feb 14, 2024Updated 2 years ago
- Convolutions and more as einsum for PyTorch☆17Jun 6, 2024Updated last year
- Voicebox: Text-Guided Multilingual Universal Speech Generation at Scale☆28Aug 4, 2023Updated 2 years ago
- A Python-based Trajectory Optimization Framework☆22Oct 29, 2024Updated last year
- ☆35Nov 22, 2024Updated last year
- Normalizing flows in PyTorch☆444Dec 1, 2025Updated 3 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 20, 2026Updated last week
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Dec 10, 2022Updated 3 years ago
- "Stochasticity in Neural ODEs: An Empirical Study". Experiments from the paper☆13Apr 27, 2020Updated 5 years ago
- Materials for a proposed Causal Inference Tutorial session at SciPy 2023☆12Jul 8, 2023Updated 2 years ago
- ☆13Oct 24, 2021Updated 4 years ago
- Experiments for Neural Flows paper☆101Nov 24, 2021Updated 4 years ago
- ☆293Sep 14, 2024Updated last year
- Improved diffusion generative models with subspaces☆135Jun 1, 2022Updated 3 years ago
- Pytorch implementation of Augmented Neural ODEs☆556Feb 9, 2023Updated 3 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆473Sep 4, 2025Updated 5 months ago
- Deep generative modeling for time-stamped heterogeneous data, enabling high-fidelity models for a large variety of spatio-temporal domain…☆107Dec 28, 2021Updated 4 years ago
- A differentiable PDE solving framework for machine learning☆1,819Feb 17, 2026Updated 2 weeks ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago