flax

一名用户演示了使用OpenAI的Codex自动生成一个Colab笔记本，该笔记本在JAX/Flax/Optax中训练一个约1000万参数的transformer进行加法运算，在T4 GPU上经过4000步后达到了高准确率。

Google Research 发布了 TimesFM 2.5，这是一个开源的时序基础模型，用于预测，拥有2亿参数、16000上下文长度，并支持最多1000步的分位数预测。该模型可在 PyPI 和 Hugging Face 上获取，支持通过 LoRA 进行微调，并集成到 Google 产品如 BigQuery ML、Google Sheets 和 Vertex Model Garden 中。