Lineax：JAX 与 Equinox 中统一的线性求解与线性最小二乘
数学软件
2023-11-30 v1
摘要
我们介绍 Lineax，一个为 JAX+Equinox 科学计算生态带来线性求解与线性最小二乘的库。Lineax 使用通用线性算子，并将线性求解与最小二乘统一为单一、可自动微分的 API。求解器与算子可由用户扩展，且无需用户实现任何自定义求导规则即可获得可微性。Lineax 可在 https://github.com/google/lineax 获取。
引用
@article{arxiv.2311.17283,
title = {Lineax: unified linear solves and linear least-squares in JAX and Equinox},
author = {Jason Rader and Terry Lyons and Patrick Kidger},
journal= {arXiv preprint arXiv:2311.17283},
year = {2023}
}
备注
7 pages, 1 figure, NeurIPS 2023 AI for Science workshop