jax框架的官方编译版本 —— 预编译发行版本(release 列表)

Angry Panda / 2024-01-17 / 原文

jax框架的Google官方给出的预编译版本,支持CUDA和CUDNN的,带有python版本号,CUDA版本号,CUDNN版本号的:

https://storage.googleapis.com/jax-releases/jax_cuda_releases.html



image