diffrax
Differential equation solver
Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX.
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
1k stars
22 watching
135 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
deep-learningdifferential-equationsdynamical-systemsequinoxjaxmachine-learningneural-differential-equationsneural-networks
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 sciml/diffeqr | Provides an R interface to solve differential equations using DifferentialEquations.jl | 143 |
 analysiscenter/pydens | An open-source framework for numerically solving ordinary and partial differential equations using neural networks | 296 |
| patrick-kidger/equinox | A JAX-based library for building and running neural networks with ease. | 2,157 |
 ucl-bug/jaxdf | A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations. | 124 |
 pbrod/numdifftools | A Python library for automatic numerical differentiation of scalar and vector-valued functions. | 258 |
| sciml/scimlbenchmarks.jl | Provides benchmarks for Scientific Machine Learning and equation solvers in multiple programming languages. | 319 |
 google-deepmind/tf2jax | Converts TensorFlow functions to equivalent JAX Python functions. | 109 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 252 |
 auto-differentiation/xad-py | Provides automatic differentiation capabilities for numerical computations in Python | 12 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 horchler/sdetools | A Matlab toolbox for numerically solving stochastic differential equations | 99 |
 coreylowman/dfdx | A deep learning library for Rust with GPU acceleration and ergonomic API. | 1,754 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 395 |
 rafavzqz/geopdes | Software package for implementing and testing isogeometric methods for solving partial differential equations | 105 |
 yhilpisch/dx | A Python-based financial analytics library for modeling and valuing complex derivatives instruments and portfolios | 719 |