optimistix

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

APACHE-2.0 License

Downloads
27K
Stars
311
Committers
7
View Code on GitHub View on X
Ecosystems: Python

Commit Statistics

Past Year

All Time

Total Commits
53
216
Total Committers
7
8
Avg. Commits Per Committer
7.57
27.0
Bot Commits
0
0

Issue Statistics

Past Year

All Time

Total Pull Requests
35
36
Merged Pull Requests
28
29
Total Issues
43
49
Time to Close Issues
9 days
9 days
Package Rankings
Top 38.5% on Pypi.org
Related Projects
lineax

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

09 May 2023 345
ninjax

General Modules for JAX

23 May 2022 57
jaxls

Sparse nonlinear least squares for JAX

16 Nov 2020 168
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
rlax

18 Feb 2020 1,193
flax

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

10 Jan 2020 5,584
optax

Optax is a gradient processing and optimization library for JAX.

12 Jun 2020 1,505
PhiML

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

06 Jun 2023 45
diffrax

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

25 Jul 2021 1,390
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
equinox

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

29 Jul 2021 1,846
dfgo

Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021

16 Aug 2021 78
jaxtyping

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

23 Jun 2022 1,131
jaxlie

Rigid transforms + Lie groups in JAX

28 Nov 2020 220