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
21 watching
130 forks
Language: Python
last commit: 3 days 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 | 141 |
 analysiscenter/pydens | An open-source framework for numerically solving ordinary and partial differential equations using neural networks | 290 |
| patrick-kidger/equinox | A JAX-based library for building and running neural networks with ease. | 2,118 |
 ucl-bug/jaxdf | A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations. | 121 |
 pbrod/numdifftools | A Python library for automatic numerical differentiation of scalar and vector-valued functions. | 256 |
| 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. | 105 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 248 |
 auto-differentiation/xad-py | Provides automatic differentiation capabilities for numerical computations in Python | 10 |
 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,737 |
| google-deepmind/dm_pix | An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research. | 389 |
 rafavzqz/geopdes | Software package for implementing and testing isogeometric methods for solving partial differential equations | 103 |
 yhilpisch/dx | A Python-based financial analytics library for modeling and valuing complex derivatives instruments and portfolios | 714 |