lineax

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

APACHE-2.0 License

Stars
345
Committers
8
View Code on GitHub View on X
Ecosystems: Python

Commit Statistics

Past Year

All Time

Total Commits
58
99
Total Committers
6
9
Avg. Commits Per Committer
9.67
11.0
Bot Commits
0
0

Issue Statistics

Past Year

All Time

Total Pull Requests
49
83
Merged Pull Requests
36
65
Total Issues
24
48
Time to Close Issues
about 1 month
19 days
Related Projects
optimistix

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

03 Oct 2023 311
optax

Optax is a gradient processing and optimization library for JAX.

12 Jun 2020 1,505
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
diffrax

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

25 Jul 2021 1,390
ninjax

General Modules for JAX

23 May 2022 57
rlax

18 Feb 2020 1,193
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816
sympy2jax

Turning SymPy expressions into JAX functions

26 Feb 2021 41
PhiML

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

06 Jun 2023 45
jaxls

Sparse nonlinear least squares for JAX

16 Nov 2020 168
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
jaxlie

Rigid transforms + Lie groups in JAX

28 Nov 2020 220
equinox

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

29 Jul 2021 1,846
flax

Flax is a neural network library for JAX that is designed for flexibility.

10 Jan 2020 5,584
jaxtyping

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

23 Jun 2022 1,131