Jax
Zhang庆欢
Acorn
展开
-
JAX如何处理返回为nan
JAX如何处理返回为nan原创 2022-08-17 18:48:38 · 281 阅读 · 1 评论 -
JAX中jit使用bug
JAX中jit只传入数据的shape,dtype,而不是数据的value原创 2022-08-17 17:06:36 · 371 阅读 · 0 评论 -
基于JAX搭建分类模型
基于JAX搭建分类模型原创 2022-08-15 17:15:19 · 420 阅读 · 0 评论 -
使用谷歌JAX遇到external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:435]
使用谷歌JAX遇到external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:435]原创 2022-08-06 17:07:49 · 1200 阅读 · 0 评论