optimistix

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

APACHE-2.0 License

Downloads

27K

Stars

311

Committers

View Code on GitHub View on X

Ecosystems: Python

Bot releases are hidden (Show)

optimistix - Initial release!

Published by patrick-kidger about 1 year ago

Hurrah! How exciting.

Package Rankings

Top 38.5% on Pypi.org

Related Projects

dm-haiku

JAX-based neural network library

18 Feb 2020 2,816

sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314

quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101

jaxls

Sparse nonlinear least squares for JAX

16 Nov 2020 168

rlax

18 Feb 2020 1,193

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

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

ninjax

General Modules for JAX

23 May 2022 57

jaxtyping

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

23 Jun 2022 1,131

diffrax

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

25 Jul 2021 1,390

lineax

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

09 May 2023 345

jaxlie

Rigid transforms + Lie groups in JAX

28 Nov 2020 220

PhiML

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

06 Jun 2023 45