equinox and optimistix
Optimistix is a specialized optimization library that builds on top of Equinox, providing nonlinear solving capabilities for users of the Equinox neural network framework—making them complements that are typically used together.
About equinox
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Equinox helps machine learning engineers and scientists build and train neural networks and scientific models using JAX. It takes your model definitions and data, then outputs trained models ready for deployment. This is for users already comfortable with JAX who need a flexible tool for advanced model building.
About optimistix
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Optimistix helps developers using JAX with advanced numerical computations by providing tools for solving complex mathematical problems like root-finding, minimization, and least squares. It takes mathematical functions and equations as input and produces precise solutions for these problems, enabling efficient development of scientific computing and machine learning applications. This tool is for JAX developers, machine learning engineers, and researchers who need robust and flexible optimization capabilities.
Scores updated daily from GitHub, PyPI, and npm data. How scores work