#非线性求解器

optimistix

Optimistix是一个基于JAX的非线性求解器库，专门用于根查找、最小化、不动点和最小二乘问题。该库提供可互操作的求解器和模块化优化器，支持PyTree状态，并与Optax兼容。Optimistix具有快速编译和运行时间，充分利用JAX的自动微分、自动并行和GPU/TPU支持等特性，为科学计算和机器学习领域提供高效工具。