产品
Flax
基于JAX的开源神经网络库，提供模块化的神经网络抽象，保留JAX函数式风格，以显式参数传递方式实现透明的多设备参数分片控制
时间轴 (近 90 天)
9月17日
DantinoX 是一个基于 JAX/Flax 的开源库，用单一模块化 Transformer 主干同时服务自回归解码、离散掩码扩散和连续流匹配三种生成范式
待验证50%
9月17日
Flax 是基于JAX的神经网络库，提供类似PyTorch Module的模块化抽象，模型参数作为显式状态传入而非隐式绑定在对象上
待验证50%
9月14日
Transformer Engine 在 JAX 生态中提供了与 JAX/Flax 集成的接口
待验证50%