#Lineax

lineax

Lineax是基于JAX开发的线性求解和最小二乘优化库，提供多种算法解决Ax = b问题。该库支持PyTree值矩阵和向量、通用线性算子及结构化矩阵，具备高效的求解器和稳定的梯度计算。Lineax优化了编译和运行性能，支持实值和复值输入，并集成JAX的自动微分、并行计算和硬件加速等功能。