Adding PyPI badge

README.md

@@ -30,8 +30,15 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 <p align="center">
     <img src="https://raw.githubusercontent.com/NVIDIA/torch-harmonics/main/images/logo/logo.png"  width="568">
+    <br>
+    <a href="https://pypi.org/project/torch_harmonics/"><img src="https://img.shields.io/pypi/v/torch_harmonics"></a>
+
 </p>
 
+<!--
+[![pypi](https://img.shields.io/pypi/v/torch_harmonics)](https://pypi.org/project/torch_harmonics/)
+-->
+
 <!-- # torch-harmonics: differentiable harmonic transforms -->
 
 <!-- ## What is torch-harmonics? -->

notebooks/train_sfno.ipynb

@@ -5,9 +5,9 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Geometric Fourier Neural Operators\n",
+    "# Spherical Fourier Neural Operators\n",
     "\n",
-    "A simple notebook to showcase geometric Fourier Neural Operators\n"
+    "A simple notebook to showcase spherical Fourier Neural Operators\n"
    ]
   },
   {