We introduce Lineax, a library bringing linear solves and linear least-squares to the JAX+Equinox scientific computing ecosystem. Lineax uses general linear operators, and unifies linear solves and least-squares into a single, autodifferentiable API. Solvers and operators are user-extensible, without requiring the user to implement any custom derivative rules to get differentiability. Lineax is available at https://github.com/google/lineax.
翻译:我们推出Lineax库,该库将线性求解与线性最小二乘法引入JAX+Equinox科学计算生态系统中。Lineax采用通用线性算子,将线性求解与最小二乘法统一为单一的可自动微分API。求解器与算子支持用户扩展,且用户无需实现任何自定义导数规则即可获得可微分性。Lineax可通过https://github.com/google/lineax获取。