Announcing "Lineax" - newest #JAX library! For fast linear solves and least squares.
GitHub: github.com/google/lineax
* Fast compile time
* Fast runtime
* Efficient new algorithms (e.g. QR) + existing ones (GMRES, LU, SVD, ...)
* Support for general linear operators🔥
GitHub: github.com/google/lineax
* Fast compile time
* Fast runtime
* Efficient new algorithms (e.g. QR) + existing ones (GMRES, LU, SVD, ...)
* Support for general linear operators🔥
Diffrax is a JAX-based library providing numerical differential equation solvers.
Features include:
1️⃣ ODE/SDE/CDE (ordinary/stochastic/controlled) solvers;
2️⃣ lots of different solvers (including Tsit5, Dopri8, symplectic solvers, implicit solvers);
3️⃣ vmappable everything (including the region of integration);
4️⃣ using a PyTree as the state;
5️⃣ dense solutions;
6️⃣ multiple adjoint methods for backpropagation;
7️⃣support for neural differential equations.
#jax
Documentation
Features include:
1️⃣ ODE/SDE/CDE (ordinary/stochastic/controlled) solvers;
2️⃣ lots of different solvers (including Tsit5, Dopri8, symplectic solvers, implicit solvers);
3️⃣ vmappable everything (including the region of integration);
4️⃣ using a PyTree as the state;
5️⃣ dense solutions;
6️⃣ multiple adjoint methods for backpropagation;
7️⃣support for neural differential equations.
#jax
Documentation
docs.kidger.site
Diffrax
The documentation for the Diffrax software library.