patrick-kidger / sympytorch
Turning SymPy expressions into PyTorch modules.
☆141Updated last year
Related projects ⓘ
Alternatives and complementary repositories for sympytorch
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆93Updated 2 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆170Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 5 months ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆269Updated 2 years ago
- ☆101Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆188Updated 4 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Matrix-free linear algebra in JAX.☆101Updated last month
- Code for efficiently sampling functions from GP(flow) posteriors☆65Updated 3 years ago
- Mathematical operations for JAX pytrees☆188Updated 5 months ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆115Updated last year
- Differentiable scientific computing library☆140Updated 2 months ago
- A library for programmatically generating equivariant layers through constraint solving☆255Updated last year
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆57Updated 4 months ago
- Initial Commit☆18Updated 3 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆330Updated last week
- Geometric kernels on manifolds, meshes and graphs☆218Updated this week
- Equation Learner, a neural network approach to symbolic regression☆71Updated last month
- A Bayesian optimization toolbox built on TensorFlow☆226Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆120Updated last month
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆244Updated this week
- A parallel ODE solver for PyTorch☆225Updated last month
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆73Updated this week
- Numerical integration of Ito or Stratonovich SDEs☆157Updated last year
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆51Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆38Updated last week
- Differentiable and numerically stable implementation of the matrix exponential☆32Updated 4 years ago
- Lie Algebra Convolutional Network implementation☆42Updated 2 years ago