Fix API spacing

docs/css/base.css

@@ -16,3 +16,7 @@
     margin-bottom: 1in;
     margin-left: 0in;
 }
+
+.md-typeset h2 {
+    line-height: 0.9;
+}

docs/css/jupyter.css

@@ -0,0 +1,7 @@
+html
+  /*
+  This adjusts the font size of Jupyter notebook code blocks to be closer to normal.
+  */
+  .highlight {
+    font-size: 85%;
+  }

docs/css/mkdocstrings.css

@@ -4,6 +4,11 @@ div.doc-contents:not(.first) {
     border-left: .15rem solid #ededed;
 }
 
+.doc-heading .highlight {
+    font-size: 18px;
+    background-color: transparent;
+}
+
 /*Mark external links as such. */
 a.external::after,
 a.autorefs-external::after {
@@ -25,3 +30,19 @@ a.external:hover::after,
 a.autorefs-external:hover::after {
     background-color: var(--md-accent-fg-color);
 }
+
+/* Fancier color for operators such as * and |. */
+.doc-signature .o {
+    color: var(--md-code-hl-special-color);
+  }
+
+/* Fancier color for constants such as None, True, and False. */
+.doc-signature .kc {
+    color: var(--md-code-hl-constant-color);
+}
+
+/* Fancier color for built-in types (only useful when cross-references are used). */
+.doc-signature .n > a[href^="https://docs.python.org/"][href*="/functions.html#"],
+.doc-signature .n > a[href^="https://docs.python.org/"][href*="/stdtypes.html#"] {
+    color: var(--md-code-hl-constant-color);
+}

mkdocs.yml

@@ -115,10 +115,11 @@ extra_css:
   - css/colors.css
   - css/mkdocstrings.css
   - css/launchy.css
+  - css/jupyter.css
 
 extra_javascript:
-  - js/mathjax-config.js
   - https://unpkg.com/mathjax@3/es5/tex-mml-chtml.js
+  - js/mathjax-config.js
 
 markdown_extensions:
   - abbr