最后发布 2024年10月4日
JAX 用于 NVIDIA GPU 的插件
JAX XLA PJRT 用于 NVIDIA GPU 的插件
JAX 的 XLA 库
微分、编译和转换 Numpy 代码。
最后发布 2024年2月26日
由以下机构支持