Jax

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

Flax 是 JAX 的神经网络库，专为灵活性而设计。

成为顶尖的人工智能与机器学习研究工程师

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

基于Kubernetes的分布式AI模型训练与微调

“How To Scale Your Model”的主页，这是一本关于在 TPU 上扩展法学硕士的简短博客式教科书

Magenta RealTime 2：一个开放权重的实时音乐模型

Django 5 实战（第5版）由 Packt 出版

Visual Novel Site

科学机器学习（SciML）基准测试、人工智能驱动的科学以及（微分）方程求解器。涵盖Julia、Python（PyTorch、Jax）、MATLAB、R

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

基于JAX的定量经济学讲座

Python+NumPy 程序的可组合转换：微分、向量化、JIT 到 GPU/TPU 等