JAX 是 Google 开发的计算机视觉研究

Posted 2021-12-07 ejinxian

        Google 开源 SCENIC，用于计算机视觉研究的 JAX 库,  来自 Google Brain 名为  的开源 JAX 库，它支持 ViT、DETR 和 MLP Mixer 等前沿的视觉模型.

        SCENIC 是用 JAX 编写的，JAX 是一种由 Google 开发的机器学习框架，专为高性能数值计算，尤其是机器学习研究而设计。它的数值函数 API 是基于 NumPy 的，NumPy 是科学计算中使用的函数集合。Python 和 NumPy 都被广泛使用和熟悉，这使得 JAX 简单、灵活且易于采用。除了 NumPy API 之外，JAX 还包括一个可扩展系统，有助于支持机器学习研究