update docs.

v0lta · v0lta · commit 122e2daeafc6 · 2022-02-24T11:15:07.000+01:00
diff --git a/docs/modules.rst b/docs/modules.rst
@@ -1,5 +1,13 @@
-ptwt
-====
+Pytorch Wavelet Toolbox (`ptwt`)
+================================
+
+The Toolbox provides padded fast wavelet transforms
+in the `conv_transform` module. The `matmul_transform` module supports boundary wavelets. 
+The `packet` module supports forward-packets transforms.
+The `sparse_math` module provides utility functions for the boundary
+wavelet transforms. Finally, the experimental `learnable_wavelets`
+module supports gradient descent optimization of wavelet filters
+with soft constraints.
 
 .. toctree::
    :maxdepth: 4
diff --git a/docs/ptwt.rst b/docs/ptwt.rst
@@ -57,10 +57,4 @@ ptwt.wavelets\_learnable module
    :undoc-members:
    :show-inheritance:
 
-Module contents
----------------
 
-.. automodule:: ptwt
-   :members:
-   :undoc-members:
-   :show-inheritance:
