jax

TimesFM是谷歌研究团队开发的预训练时间序列基础模型，专注于时间序列预测任务。

一个基于MLIR的可重定向机器学习编译器与运行时工具包。

基于JAX的高效可微分n维偏微分方程求解器。