PEP 503仓库索引，用于jax[cuda]

每日自动重建

仓库

https://jorenham.github.io/jax_pep503/

示例：将jax[cuda]添加到Poetry项目中

• 将仓库作为二级源添加到您的pyproject.toml

  ...
  
  [[tool.poetry.source]]
  name = "PyPI"
  priority = "primary"
  
  [[tool.poetry.source]]
  name = "jorenham/jax_pep503"
  url = "https://jorenham.github.io/jax_pep503/"
  priority = "supplemental"
  
  [tool.poetry.dependencies]
  ...
  

• 现在您可以使用以下命令添加jax[cuda]

  poetry add jax[cuda]
  

有关更多详细信息，请参阅示例项目。

另请参阅

• PEP 503 – 简单仓库API
• google/jax
• google/jax#5410 – 请为包的CUDA版本提供PEP 503兼容索引
• jax-releases (不兼容PEP 503)