最近发布 2024年10月4日
NVIDIA GPU 的 JAX 插件
JAX XLA PJRT 插件用于 NVIDIA GPU
JAX 的 XLA 库
对 Numpy 代码进行微分、编译和转换。
最近发布 2024年9月13日
无
支持者
