jax与jaxlib安装记录

jax与jaxlib 安装记录

  • jax版本要与jaxlib版本一致
  • win安装jaxlib不能直接用pip安装,网络上有非官方的编译版本
  • jaxlib的版本选择要与Python版本保持一致

你可能感兴趣的:(python)