diff --git a/docs/src/index.rst b/docs/src/index.rst index 51e719572..76907dbe0 100644 --- a/docs/src/index.rst +++ b/docs/src/index.rst @@ -70,6 +70,7 @@ are the CPU and GPU. python/fft python/linalg python/metal + python/cuda python/memory_management python/nn python/optimizers