@@ -497,7 +497,7 @@ <h1>LowRankAffineEigenvaluePMM<a class="headerlink" href="#lowrankaffineeigenval
497497< p > < code class ="docutils literal notranslate "> < span class ="pre "> parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM</ span > </ code > </ p >
498498< dl class ="py class ">
499499< dt class ="sig sig-object py " id ="parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM ">
500- < em class ="property "> < span class ="k "> < span class ="pre "> class</ span > </ span > < span class ="w "> </ span > </ em > < span class ="sig-name descname "> < span class ="pre "> LowRankAffineEigenvaluePMM</ span > </ span > < span class ="sig-paren "> (</ span > < em class ="sig-param "> < span class ="n "> < span class ="pre "> matrix_size</ 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 "> rank</ 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 "> num_eig</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 1</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> which</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 'SA'</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> smoothing</ 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 "> us</ 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 "> init_magnitude</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 0.01</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> bias_term</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> True</ span > </ span > </ em > < span class ="sig-paren "> )</ span > < a class ="reference internal " href ="../_modules/parametricmatrixmodels/modules/lowrankaffineeigenvaluepmm.html#LowRankAffineEigenvaluePMM "> < span class ="viewcode-link "> < span class ="pre "> [source]</ span > </ span > </ a > < a class ="headerlink " href ="#parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM " title ="Link to this definition "> #</ a > </ dt >
500+ < em class ="property "> < span class ="k "> < span class ="pre "> class</ span > </ span > < span class ="w "> </ span > </ em > < span class ="sig-name descname "> < span class ="pre "> LowRankAffineEigenvaluePMM</ span > </ span > < span class ="sig-paren "> (</ span > < em class ="sig-param "> < span class ="n "> < span class ="pre "> matrix_size</ 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 "> rank</ 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 "> num_eig</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 1</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> which</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 'SA'</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> smoothing</ 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 "> lambdas</ 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 "> us</ 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 "> init_magnitude</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 0.01</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> bias_term</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> True</ span > </ span > </ em > < span class ="sig-paren "> )</ span > < a class ="reference internal " href ="../_modules/parametricmatrixmodels/modules/lowrankaffineeigenvaluepmm.html#LowRankAffineEigenvaluePMM "> < span class ="viewcode-link "> < span class ="pre "> [source]</ span > </ span > </ a > < a class ="headerlink " href ="#parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM " title ="Link to this definition "> #</ a > </ dt >
501501< dd > < p > Bases: < a class ="reference internal " href ="parametricmatrixmodels.modules.MultiModule.html#parametricmatrixmodels.modules.MultiModule " title ="parametricmatrixmodels.modules.multimodule.MultiModule "> < code class ="xref py py-class docutils literal notranslate "> < span class ="pre "> MultiModule</ span > </ code > </ a > </ p >
502502< p > < code class ="docutils literal notranslate "> < span class ="pre "> LowRankAffineEigenvaluePMM</ span > </ code > is a module that implements the affine
503503eigenvalue Parametric Matrix Model (PMM) using low-rank matrices via
@@ -542,7 +542,7 @@ <h1>LowRankAffineEigenvaluePMM<a class="headerlink" href="#lowrankaffineeigenval
542542</ aside >
543543< dl class ="py method ">
544544< dt class ="sig sig-object py " id ="parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM.__init__ ">
545- < span class ="sig-name descname "> < span class ="pre "> __init__</ span > </ span > < span class ="sig-paren "> (</ span > < em class ="sig-param "> < span class ="n "> < span class ="pre "> matrix_size</ 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 "> rank</ 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 "> num_eig</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 1</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> which</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 'SA'</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> smoothing</ 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 "> us</ 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 "> init_magnitude</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 0.01</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> bias_term</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> True</ span > </ span > </ em > < span class ="sig-paren "> )</ span > < a class ="reference internal " href ="../_modules/parametricmatrixmodels/modules/lowrankaffineeigenvaluepmm.html#LowRankAffineEigenvaluePMM.__init__ "> < span class ="viewcode-link "> < span class ="pre "> [source]</ span > </ span > </ a > < a class ="headerlink " href ="#parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM.__init__ " title ="Link to this definition "> #</ a > </ dt >
545+ < span class ="sig-name descname "> < span class ="pre "> __init__</ span > </ span > < span class ="sig-paren "> (</ span > < em class ="sig-param "> < span class ="n "> < span class ="pre "> matrix_size</ 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 "> rank</ 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 "> num_eig</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 1</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> which</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 'SA'</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> smoothing</ 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 "> lambdas</ 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 "> us</ 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 "> init_magnitude</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> 0.01</ span > </ span > </ em > , < em class ="sig-param "> < span class ="n "> < span class ="pre "> bias_term</ span > </ span > < span class ="o "> < span class ="pre "> =</ span > </ span > < span class ="default_value "> < span class ="pre "> True</ span > </ span > </ em > < span class ="sig-paren "> )</ span > < a class ="reference internal " href ="../_modules/parametricmatrixmodels/modules/lowrankaffineeigenvaluepmm.html#LowRankAffineEigenvaluePMM.__init__ "> < span class ="viewcode-link "> < span class ="pre "> [source]</ span > </ span > </ a > < a class ="headerlink " href ="#parametricmatrixmodels.modules.LowRankAffineEigenvaluePMM.__init__ " title ="Link to this definition "> #</ a > </ dt >
546546< dd > < p > Initialize the < code class ="docutils literal notranslate "> < span class ="pre "> AffineEigenvaluePMM</ span > </ code > module.</ p >
547547< p > By default this module is initialized to compute the smallest algebraic
548548eigenvalue (ground state).</ p >
@@ -571,6 +571,12 @@ <h1>LowRankAffineEigenvaluePMM<a class="headerlink" href="#lowrankaffineeigenval
571571</ p > </ li >
572572< li > < p > < strong > smoothing</ strong > (< span class ="sphinx_autodoc_typehints-type "> < a class ="reference external " href ="https://docs.python.org/3/library/functions.html#float " title ="(in Python v3.13) "> < code class ="xref py py-class docutils literal notranslate "> < span class ="pre "> float</ span > </ code > </ a > </ span > ) – Optional smoothing parameter. Set to < code class ="docutils literal notranslate "> < span class ="pre "> 0.0</ span > </ code > to disable
573573smoothing. Default is < code class ="docutils literal notranslate "> < span class ="pre "> None</ span > </ code > /< code class ="docutils literal notranslate "> < span class ="pre "> 0.0</ span > </ code > (no smoothing).</ p > </ li >
574+ < li > < p > < strong > lambdas</ strong > (< span class ="sphinx_autodoc_typehints-type "> < a class ="reference external " href ="https://docs.jax.dev/en/latest/_autosummary/jax.Array.html#jax.Array " title ="(in JAX) "> < code class ="xref py py-class docutils literal notranslate "> < span class ="pre "> Array</ span > </ code > </ a > </ span > ) – Optional array of shape < cite > (input_size+1, rank)</ cite > (if
575+ < code class ="docutils literal notranslate "> < span class ="pre "> bias_term</ span > </ code > is < code class ="docutils literal notranslate "> < span class ="pre "> True</ span > </ code > ) or < cite > (input_size, rank)</ cite > (if
576+ < code class ="docutils literal notranslate "> < span class ="pre "> bias_term</ span > </ code > is < code class ="docutils literal notranslate "> < span class ="pre "> False</ span > </ code > ), containing the < cite > lambda_k^i</ cite > real
577+ coefficients used to construct the low-rank Hermitian matrices.
578+ If not provided, the coefficients will be initialized randomly
579+ when the module is compiled.</ p > </ li >
574580< li > < p > < strong > us</ strong > (< span class ="sphinx_autodoc_typehints-type "> < a class ="reference external " href ="https://docs.jax.dev/en/latest/_autosummary/jax.Array.html#jax.Array " title ="(in JAX) "> < code class ="xref py py-class docutils literal notranslate "> < span class ="pre "> Array</ span > </ code > </ a > </ span > ) – Optional array of shape
575581< code class ="docutils literal notranslate "> < span class ="pre "> (input_size+1,</ span > < span class ="pre "> rank,</ span > < span class ="pre "> matrix_size)</ span > </ code > (if < code class ="docutils literal notranslate "> < span class ="pre "> bias_term</ span > </ code >
576582is < code class ="docutils literal notranslate "> < span class ="pre "> True</ span > </ code > ) or < code class ="docutils literal notranslate "> < span class ="pre "> (input_size,</ span > < span class ="pre "> rank,</ span > < span class ="pre "> matrix_size)</ span > </ code > (if
0 commit comments