diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

APACHE-2.0 License

Downloads
26.6K
Stars
1.4K
Committers
28
View Code on GitHub View on X
Ecosystems: Python

Commit Statistics

Past Year

All Time

Total Commits
91
543
Total Committers
13
28
Avg. Commits Per Committer
7.0
19.39
Bot Commits
0
0

Issue Statistics

Past Year

All Time

Total Pull Requests
75
124
Merged Pull Requests
51
85
Total Issues
112
232
Time to Close Issues
6 days
about 1 month
Package Rankings
Top 6.68% on Proxy.golang.org
Top 30.98% on Conda-forge.org
Top 2.94% on Pypi.org
Related Projects
optax

Optax is a gradient processing and optimization library for JAX.

12 Jun 2020 1,505
DifferentialEquations.jl

Multi-language suite for high-performance solvers of differential equations and scientific machin...

11 May 2016 2,853
optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.sit...

03 Oct 2023 311
jaxls

Sparse nonlinear least squares for JAX

16 Nov 2020 168
finitediffX

Finite difference tools in JAX

01 Mar 2023 8
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...

23 Jun 2022 1,131
tensorf-jax

Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)

17 Apr 2022 39
sympy2jax

Turning SymPy expressions into JAX functions

26 Feb 2021 41
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
PhiML

Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy

06 Jun 2023 45
dfgo

Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021

16 Aug 2021 78
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
jaxlie

Rigid transforms + Lie groups in JAX

28 Nov 2020 220