diff --git a/README.md b/README.md index 20fb6c5c..6cbf76c2 100644 --- a/README.md +++ b/README.md @@ -16,7 +16,7 @@ pip install cuequivariance-jax pip install cuequivariance-torch pip install cuequivariance # Installs only the core non-ML components -# CUDA kernels for different CUDA versions +# CUDA kernels pip install cuequivariance-ops-jax-cu12 pip install cuequivariance-ops-torch-cu12 # or cu11 ``` diff --git a/docs/changelog.md b/docs/changelog.md index e1a84149..c38e409a 100644 --- a/docs/changelog.md +++ b/docs/changelog.md @@ -1,3 +1,3 @@ -# Changelog +# Release Notes ```{include} ../CHANGELOG.md diff --git a/docs/index.rst b/docs/index.rst index 6468a5b4..2de84a60 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -39,14 +39,16 @@ The easiest way to install cuEquivariance is from `PyPi `_ us pip install cuequivariance-torch pip install cuequivariance # Installs only the core non-ML components - # CUDA kernels for different CUDA versions + # CUDA kernels pip install cuequivariance-ops-torch-cu11 pip install cuequivariance-ops-torch-cu12 + pip install cuequivariance-ops-jax-cu12 Requirements ------------ -``cuequivariance-ops-torch-*`` packages are only available for Linux x86_64 and require PyTorch 2.4.0 or later. + - ``cuequivariance-ops-torch-*`` packages are available for Linux x86_64/aarch64 and require PyTorch 2.4.0 or later. aarch64 is only available for Python 3.12. + - ``cuequivariance-ops-jax-cu12`` package is only available for Linux x86_64 and requires JAX 0.5.0 or later. Organization ------------