Deploying to gh-pages from @ cd238cf 🚀

.buildinfo

@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: c5370ba81785d8a03b9d289bde176125
+config: 7d6ef3275af9ac50eddf23518bdab2d1
 tags: 645f666f9bcd5a90fca523b33c5a78b7

_static/copybutton.js

@@ -20,7 +20,7 @@ const messages = {
   },
   'fr' : {
     'copy': 'Copier',
-    'copy_to_clipboard': 'Copié dans le presse-papier',
+    'copy_to_clipboard': 'Copier dans le presse-papier',
     'copy_success': 'Copié !',
     'copy_failure': 'Échec de la copie',
   },
@@ -224,7 +224,7 @@ var copyTargetText = (trigger) => {
   var target = document.querySelector(trigger.attributes['data-clipboard-target'].value);
 
   // get filtered text
-  let exclude = '.linenos, .gp';
+  let exclude = '.linenos';
 
   let text = filterText(target, exclude);
   return formatCopyText(text, '', false, true, true, true, '', '')

_static/nbsphinx-code-cells.css

@@ -62,12 +62,16 @@ div.nblast.container {
 }
 
 /* input prompt */
-div.nbinput.container div.prompt pre {
+div.nbinput.container div.prompt pre,
+/* for sphinx_immaterial theme: */
+div.nbinput.container div.prompt pre > code {
     color: #307FC1;
 }
 
 /* output prompt */
-div.nboutput.container div.prompt pre {
+div.nboutput.container div.prompt pre,
+/* for sphinx_immaterial theme: */
+div.nboutput.container div.prompt pre > code {
     color: #BF5B3D;
 }
 
@@ -204,8 +208,10 @@ div.nboutput.container div.output_area > div[class^='highlight']{
     overflow-y: hidden;
 }
 
-/* hide copybtn icon on prompts (needed for 'sphinx_copybutton') */
-.prompt .copybtn {
+/* hide copy button on prompts for 'sphinx_copybutton' extension ... */
+.prompt .copybtn,
+/* ... and 'sphinx_immaterial' theme */
+.prompt .md-clipboard.md-icon {
     display: none;
 }
 

_static/pygments.css

@@ -22,6 +22,7 @@ html[data-theme="light"] .highlight .c1 { color: #8f5902; font-style: italic } /
 html[data-theme="light"] .highlight .cs { color: #8f5902; font-style: italic } /* Comment.Special */
 html[data-theme="light"] .highlight .gd { color: #a40000 } /* Generic.Deleted */
 html[data-theme="light"] .highlight .ge { color: #000000; font-style: italic } /* Generic.Emph */
+html[data-theme="light"] .highlight .ges { color: #000000; font-weight: bold; font-style: italic } /* Generic.EmphStrong */
 html[data-theme="light"] .highlight .gr { color: #ef2929 } /* Generic.Error */
 html[data-theme="light"] .highlight .gh { color: #000080; font-weight: bold } /* Generic.Heading */
 html[data-theme="light"] .highlight .gi { color: #00A000 } /* Generic.Inserted */
@@ -88,35 +89,36 @@ html[data-theme="dark"] .highlight td.linenos .special { color: #000000; backgro
 html[data-theme="dark"] .highlight span.linenos.special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
 html[data-theme="dark"] .highlight .hll { background-color: #49483e }
 html[data-theme="dark"] .highlight { background: #272822; color: #f8f8f2 }
-html[data-theme="dark"] .highlight .c { color: #75715e } /* Comment */
-html[data-theme="dark"] .highlight .err { color: #960050; background-color: #1e0010 } /* Error */
+html[data-theme="dark"] .highlight .c { color: #959077 } /* Comment */
+html[data-theme="dark"] .highlight .err { color: #ed007e; background-color: #1e0010 } /* Error */
 html[data-theme="dark"] .highlight .esc { color: #f8f8f2 } /* Escape */
 html[data-theme="dark"] .highlight .g { color: #f8f8f2 } /* Generic */
 html[data-theme="dark"] .highlight .k { color: #66d9ef } /* Keyword */
 html[data-theme="dark"] .highlight .l { color: #ae81ff } /* Literal */
 html[data-theme="dark"] .highlight .n { color: #f8f8f2 } /* Name */
-html[data-theme="dark"] .highlight .o { color: #f92672 } /* Operator */
+html[data-theme="dark"] .highlight .o { color: #ff4689 } /* Operator */
 html[data-theme="dark"] .highlight .x { color: #f8f8f2 } /* Other */
 html[data-theme="dark"] .highlight .p { color: #f8f8f2 } /* Punctuation */
-html[data-theme="dark"] .highlight .ch { color: #75715e } /* Comment.Hashbang */
-html[data-theme="dark"] .highlight .cm { color: #75715e } /* Comment.Multiline */
-html[data-theme="dark"] .highlight .cp { color: #75715e } /* Comment.Preproc */
-html[data-theme="dark"] .highlight .cpf { color: #75715e } /* Comment.PreprocFile */
-html[data-theme="dark"] .highlight .c1 { color: #75715e } /* Comment.Single */
-html[data-theme="dark"] .highlight .cs { color: #75715e } /* Comment.Special */
-html[data-theme="dark"] .highlight .gd { color: #f92672 } /* Generic.Deleted */
+html[data-theme="dark"] .highlight .ch { color: #959077 } /* Comment.Hashbang */
+html[data-theme="dark"] .highlight .cm { color: #959077 } /* Comment.Multiline */
+html[data-theme="dark"] .highlight .cp { color: #959077 } /* Comment.Preproc */
+html[data-theme="dark"] .highlight .cpf { color: #959077 } /* Comment.PreprocFile */
+html[data-theme="dark"] .highlight .c1 { color: #959077 } /* Comment.Single */
+html[data-theme="dark"] .highlight .cs { color: #959077 } /* Comment.Special */
+html[data-theme="dark"] .highlight .gd { color: #ff4689 } /* Generic.Deleted */
 html[data-theme="dark"] .highlight .ge { color: #f8f8f2; font-style: italic } /* Generic.Emph */
+html[data-theme="dark"] .highlight .ges { color: #f8f8f2; font-weight: bold; font-style: italic } /* Generic.EmphStrong */
 html[data-theme="dark"] .highlight .gr { color: #f8f8f2 } /* Generic.Error */
 html[data-theme="dark"] .highlight .gh { color: #f8f8f2 } /* Generic.Heading */
 html[data-theme="dark"] .highlight .gi { color: #a6e22e } /* Generic.Inserted */
 html[data-theme="dark"] .highlight .go { color: #66d9ef } /* Generic.Output */
-html[data-theme="dark"] .highlight .gp { color: #f92672; font-weight: bold } /* Generic.Prompt */
+html[data-theme="dark"] .highlight .gp { color: #ff4689; font-weight: bold } /* Generic.Prompt */
 html[data-theme="dark"] .highlight .gs { color: #f8f8f2; font-weight: bold } /* Generic.Strong */
-html[data-theme="dark"] .highlight .gu { color: #75715e } /* Generic.Subheading */
+html[data-theme="dark"] .highlight .gu { color: #959077 } /* Generic.Subheading */
 html[data-theme="dark"] .highlight .gt { color: #f8f8f2 } /* Generic.Traceback */
 html[data-theme="dark"] .highlight .kc { color: #66d9ef } /* Keyword.Constant */
 html[data-theme="dark"] .highlight .kd { color: #66d9ef } /* Keyword.Declaration */
-html[data-theme="dark"] .highlight .kn { color: #f92672 } /* Keyword.Namespace */
+html[data-theme="dark"] .highlight .kn { color: #ff4689 } /* Keyword.Namespace */
 html[data-theme="dark"] .highlight .kp { color: #66d9ef } /* Keyword.Pseudo */
 html[data-theme="dark"] .highlight .kr { color: #66d9ef } /* Keyword.Reserved */
 html[data-theme="dark"] .highlight .kt { color: #66d9ef } /* Keyword.Type */
@@ -135,9 +137,9 @@ html[data-theme="dark"] .highlight .nl { color: #f8f8f2 } /* Name.Label */
 html[data-theme="dark"] .highlight .nn { color: #f8f8f2 } /* Name.Namespace */
 html[data-theme="dark"] .highlight .nx { color: #a6e22e } /* Name.Other */
 html[data-theme="dark"] .highlight .py { color: #f8f8f2 } /* Name.Property */
-html[data-theme="dark"] .highlight .nt { color: #f92672 } /* Name.Tag */
+html[data-theme="dark"] .highlight .nt { color: #ff4689 } /* Name.Tag */
 html[data-theme="dark"] .highlight .nv { color: #f8f8f2 } /* Name.Variable */
-html[data-theme="dark"] .highlight .ow { color: #f92672 } /* Operator.Word */
+html[data-theme="dark"] .highlight .ow { color: #ff4689 } /* Operator.Word */
 html[data-theme="dark"] .highlight .pm { color: #f8f8f2 } /* Punctuation.Marker */
 html[data-theme="dark"] .highlight .w { color: #f8f8f2 } /* Text.Whitespace */
 html[data-theme="dark"] .highlight .mb { color: #ae81ff } /* Literal.Number.Bin */

api/precompute_transforms/construct.html

@@ -461,6 +461,37 @@
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="s2fft.precompute_transforms.construct.spin_spherical_kernel_jax">
+<span class="sig-prename descclassname"><span class="pre">s2fft.precompute_transforms.construct.</span></span><span class="sig-name descname"><span class="pre">spin_spherical_kernel_jax</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">L</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">spin</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">reality</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">sampling</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">str</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">'mw'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">nside</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">forward</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#s2fft.precompute_transforms.construct.spin_spherical_kernel_jax" title="Permalink to this definition"></a></dt>
+<dd><p>Precompute the wigner-d kernel for spin-spherical transform. This can be
+drastically faster but comes at a <span class="math notranslate nohighlight">\(\mathcal{O}(L^3)\)</span> memory overhead, making
+it infeasible for <span class="math notranslate nohighlight">\(L\geq 512\)</span>.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters</dt>
+<dd class="field-odd"><ul class="simple">
+<li><p><strong>L</strong> (<em>int</em>) – Harmonic band-limit.</p></li>
+<li><p><strong>spin</strong> (<em>int</em>) – Harmonic spin.</p></li>
+<li><p><strong>reality</strong> (<em>bool</em><em>, </em><em>optional</em>) – Whether the signal on the sphere is real.  If so,
+conjugate symmetry is exploited to reduce computational costs.
+Defaults to False.</p></li>
+<li><p><strong>sampling</strong> (<em>str</em><em>, </em><em>optional</em>) – Sampling scheme.  Supported sampling schemes include
+{“mw”, “mwss”, “dh”, “healpix”}. Defaults to “mw”.</p></li>
+<li><p><strong>nside</strong> (<em>int</em>) – HEALPix Nside resolution parameter.  Only required
+if sampling=”healpix”.</p></li>
+<li><p><strong>forward</strong> (<em>bool</em><em>, </em><em>optional</em>) – Whether to provide forward or inverse shift.
+Defaults to False.</p></li>
+</ul>
+</dd>
+<dt class="field-even">Returns</dt>
+<dd class="field-even"><p>Transform kernel for spin-spherical harmonic transform.</p>
+</dd>
+<dt class="field-odd">Return type</dt>
+<dd class="field-odd"><p>jnp.ndarray</p>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="s2fft.precompute_transforms.construct.wigner_kernel">
 <span class="sig-prename descclassname"><span class="pre">s2fft.precompute_transforms.construct.</span></span><span class="sig-name descname"><span class="pre">wigner_kernel</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">L</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">N</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">reality</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">sampling</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">str</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">'mw'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">nside</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">forward</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#s2fft.precompute_transforms.construct.wigner_kernel" title="Permalink to this definition"></a></dt>
@@ -492,6 +523,37 @@
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="s2fft.precompute_transforms.construct.wigner_kernel_jax">
+<span class="sig-prename descclassname"><span class="pre">s2fft.precompute_transforms.construct.</span></span><span class="sig-name descname"><span class="pre">wigner_kernel_jax</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">L</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">N</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">int</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">reality</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">sampling</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">str</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">'mw'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">nside</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">forward</span></span><span class="p"><span class="pre">:</span></span> <span class="n"><span class="pre">bool</span></span> <span class="o"><span class="pre">=</span></span> <span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#s2fft.precompute_transforms.construct.wigner_kernel_jax" title="Permalink to this definition"></a></dt>
+<dd><p>Precompute the wigner-d kernels required for a Wigner transform. This can be
+drastically faster but comes at a <span class="math notranslate nohighlight">\(\mathcal{O}(NL^3)\)</span> memory overhead, making
+it infeasible for <span class="math notranslate nohighlight">\(L \geq 512\)</span>.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters</dt>
+<dd class="field-odd"><ul class="simple">
+<li><p><strong>L</strong> (<em>int</em>) – Harmonic band-limit.</p></li>
+<li><p><strong>N</strong> (<em>int</em>) – Directional band-limit.</p></li>
+<li><p><strong>reality</strong> (<em>bool</em><em>, </em><em>optional</em>) – Whether the signal on the sphere is real.  If so,
+conjugate symmetry is exploited to reduce computational costs.
+Defaults to False.</p></li>
+<li><p><strong>sampling</strong> (<em>str</em><em>, </em><em>optional</em>) – Sampling scheme.  Supported sampling schemes include
+{“mw”, “mwss”, “dh”, “healpix”}. Defaults to “mw”.</p></li>
+<li><p><strong>nside</strong> (<em>int</em>) – HEALPix Nside resolution parameter.  Only required
+if sampling=”healpix”.</p></li>
+<li><p><strong>forward</strong> (<em>bool</em><em>, </em><em>optional</em>) – Whether to provide forward or inverse shift.
+Defaults to False.</p></li>
+</ul>
+</dd>
+<dt class="field-even">Returns</dt>
+<dd class="field-even"><p>Transform kernel for Wigner transform.</p>
+</dd>
+<dt class="field-odd">Return type</dt>
+<dd class="field-odd"><p>jnp.ndarray</p>
+</dd>
+</dl>
+</dd></dl>
+
 </div>