diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
AI summary
Differential equation solver
Provides numerical differential equation solvers using autodifferentiable and GPU-capable JAX.
- stars
- 1.5K
- forks
- 135
- watching
- 22
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Differential equation solver
Provides an R interface to solve differential equations using DifferentialEquations.jl
PDE solver
An open-source framework for numerically solving ordinary and partial differential equations using neural networks
Neural network library
A JAX-based library for building and running neural networks with ease.
Numerical simulator framework
A JAX-based framework for creating differentiable numerical simulators with arbitrary discretizations.
Derivative calculator
A Python library for automatic numerical differentiation of scalar and vector-valued functions.
Equation solver benchmarks
Provides benchmarks for Scientific Machine Learning and equation solvers in multiple programming languages.
Converter
Converts TensorFlow functions to equivalent JAX Python functions.
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
Derivatives calculator
Provides automatic differentiation capabilities for numerical computations in Python
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
SDE solver
A Matlab toolbox for numerically solving stochastic differential equations
coreylowman/dfdx1.8K
Deep learning library
A deep learning library for Rust with GPU acceleration and ergonomic API.
Image processor
An image processing library built on top of JAX to provide optimized and parallelized functions for machine learning research.
PDE solver
Software package for implementing and testing isogeometric methods for solving partial differential equations
yhilpisch/dx719
Derivatives analyzer
A Python-based financial analytics library for modeling and valuing complex derivatives instruments and portfolios
Featured in 1 awesome list
Each link jumps to the spot where the list mentions diffrax.
