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

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

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



image

posted on 2024-01-17 18:36  Angry_Panda  阅读(222)  评论(0)    收藏  举报

导航