Fix links in readthedocs; update PyPI management

PiperOrigin-RevId: 392519336

neural_tangents/stax.py

@@ -3480,24 +3480,24 @@ def ImageResize(
   The `method` argument expects one of the following resize methods:
 
   `ResizeMethod.NEAREST`, `"nearest"`:
-    Nearest neighbor interpolation_. The values of `antialias` and `precision`
+    `Nearest neighbor interpolation`_. The values of `antialias` and `precision`
     are ignored.
 
   `ResizeMethod.LINEAR`, `"linear"`, `"bilinear"`, `"trilinear"`, `"triangle"`:
-    Linear interpolation_. If `antialias` is ``True``, uses a triangular filter
-    when downsampling.
+    `Linear interpolation`_. If `antialias` is ``True``, uses a triangular
+    filter when downsampling.
 
   The following methods are NOT SUPPORTED in `kernel_fn` (only `init_fn` and
   `apply_fn` work):
 
   `ResizeMethod.CUBIC`, `"cubic"`, `"bicubic"`, `"tricubic"`:
-    Cubic interpolation_, using the Keys cubic kernel.
+    `Cubic interpolation`_, using the Keys cubic kernel.
 
   `ResizeMethod.LANCZOS3`, `"lanczos3"`:
-    Lanczos resampling_, using a kernel of radius 3.
+    `Lanczos resampling`_, using a kernel of radius 3.
 
   `ResizeMethod.LANCZOS5`, `"lanczos5"`:
-    Lanczos resampling_, using a kernel of radius 5.
+    `Lanczos resampling`_, using a kernel of radius 5.
 
   .. _Nearest neighbor interpolation: https://en.wikipedia.org/wiki/Nearest-neighbor_interpolation
   .. _Linear interpolation: https://en.wikipedia.org/wiki/Bilinear_interpolation

setup.py

@@ -17,8 +17,6 @@
 
 
 import os
-import sys
-
 import setuptools
 
 
@@ -58,15 +56,20 @@ def _get_version() -> str:
     name='neural-tangents',
     version=_get_version(),
     license='Apache 2.0',
-    author='Google',
-    author_email='neural-tangents-dev@google.com',
+    author='Neural Tangents developers',
+    author_email='nt@google.com',
     install_requires=INSTALL_REQUIRES,
     url='https://github.com/google/neural-tangents',
     download_url='https://pypi.org/project/neural-tangents/',
     project_urls={
         'Source Code': 'https://github.com/google/neural-tangents',
-        'Documentation': 'https://arxiv.org/abs/1912.02803',
+        'Paper': 'https://arxiv.org/abs/1912.02803',
+        'Video': 'https://iclr.cc/virtual_2020/poster_SklD9yrFPS.html',
+        'Documentation': 'https://neural-tangents.readthedocs.io/en/latest/?badge=latest',
         'Bug Tracker': 'https://github.com/google/neural-tangents/issues',
+        'Release Notes': 'https://github.com/google/neural-tangents/releases',
+        'PyPi': 'https://pypi.org/project/neural-tangents/',
+        'Tests': 'https://travis-ci.org/github/google/neural-tangents',
     },
     packages=setuptools.find_packages(exclude=('presentation',)),
     long_description=long_description,