[P] Optimistix, nonlinear optimisation in JAX+Equinox!

• optimistix

  2 237 8.5 Python

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

Hi everyone! I wanted to advertise my new JAX optimisation library Optimistix!

• diffrax

  21 1,230 8.3 Python

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

Optimistix has high-level APIs for minimisation, least-squares, root-finding, and fixed-point iteration and was written to take care of these kinds of subroutines in Diffrax.

• InfluxDB

  www.influxdata.com sponsored

  Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  

• equinox

  31 1,809 9.2 Python

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

The elevator pitch is Optimistix is really fast, especially to compile. It plays nicely with Optax for first-order gradient-based methods, and takes a lot of design inspiration from Equinox, representing the state of all the solvers as standard JAX PyTrees.

Suggest a related project