Index

_

__init__() (bijx.AffineLinear method) (bijx.ApplyBijection method) (bijx.ArrayDistribution method) (bijx.AutoJacVF method) (bijx.BetaStretch method) (bijx.Bijection method) (bijx.BinaryMask method) (bijx.BufferedSampler method) (bijx.Chain method) (bijx.CondInverse method) (bijx.Const method) (bijx.ContFlowCG method) (bijx.ContFlowDiffrax method) (bijx.ContFlowRK4 method) (bijx.ConvVF method) (bijx.DiagonalNormal method) (bijx.DiffraxConfig method) (bijx.Distribution method) (bijx.ExpandDims method) (bijx.Exponential method) (bijx.FreeTheoryScaling method) (bijx.Frozen method) (bijx.GaussianCDF method) (bijx.GaussianMixture method) (bijx.GeneralCouplingLayer method) (bijx.IMH method) (bijx.IMHInfo method) (bijx.IMHState method) (bijx.IndependentNormal method) (bijx.IndependentUniform method) (bijx.Inverse method) (bijx.MetaLayer method) (bijx.MixtureStack method) (bijx.ModuleReconstructor method) (bijx.MonotoneRQSpline method) (bijx.MultivariateNormal method) (bijx.Partial method) (bijx.Power method) (bijx.Reshape method) (bijx.ScalarBijection method) (bijx.Scaling method) (bijx.ScanChain method) (bijx.ShapeInfo method) (bijx.Shift method) (bijx.Sigmoid method) (bijx.Sinh method) (bijx.SoftPlus method) (bijx.SpectrumScaling method) (bijx.SqueezeDims method) (bijx.Tan method) (bijx.Tanh method) (bijx.ToFourierData method) (bijx.Transformed method)

A

a (bijx.cg.ButcherTableau attribute) accept_prob (bijx.IMHInfo attribute) adjoint (bijx.DiffraxConfig attribute) adjoint() (in module bijx.lie) AffineLinear (class in bijx) apply() (bijx.ApplyBijection method) (bijx.MonotoneRQSpline method) (bijx.SpectrumScaling method)

apply_feature_map() (bijx.nn.features.ConcatFeatures method) (bijx.nn.features.FourierFeatures method) (bijx.nn.features.NonlinearFeatures method) (bijx.nn.features.PolynomialFeatures method) apply_gauge_sym() (in module bijx.lattice.gauge) ApplyBijection (class in bijx) ArrayDistribution (class in bijx) AutoJacVF (class in bijx)

B

b (bijx.cg.ButcherTableau attribute) BetaStretch (class in bijx) Bijection (class in bijx) bijx module bijx.cg module bijx.fourier module bijx.lattice module bijx.lattice.gauge module bijx.lattice.scalar module

bijx.lie module bijx.nn.conv module bijx.nn.embeddings module bijx.nn.features module bijx.nn.nets module BinaryMask (class in bijx) boolean_mask (bijx.BinaryMask property) BufferedSampler (class in bijx) build() (bijx.ConvVF class method) ButcherTableau (class in bijx.cg)

C

c (bijx.cg.ButcherTableau attribute) CG2 (in module bijx.cg) CG3 (in module bijx.cg) cg_stage() (in module bijx.cg) Chain (class in bijx) channel_axes (bijx.ShapeInfo property) channel_dim (bijx.ShapeInfo attribute) channel_shape (bijx.ShapeInfo attribute) channel_size (bijx.ShapeInfo property) channel_slices (bijx.fourier.FourierMeta property) checker_mask() (in module bijx) comp_complex (bijx.fourier.FFTRep attribute) comp_real (bijx.fourier.FFTRep attribute) complex_to_rdof() (bijx.fourier.FourierData static method) complex_to_rfft() (bijx.fourier.FourierData static method) compute_haar_density() (in module bijx.lie) ConcatFeatures (class in bijx.nn.features) CondInverse (class in bijx) Const (class in bijx) construct_su_matrix_from_eigenvalues() (in module bijx.lie) ContFlowCG (class in bijx) ContFlowDiffrax (class in bijx) ContFlowRK4 (class in bijx)

contract() (in module bijx.lie) controller_state (bijx.DiffraxConfig attribute) conv_indices() (in module bijx.nn.conv) ConvNet (class in bijx.nn.nets) ConvSym (class in bijx.nn.conv) ConvVF (class in bijx) copy_from (bijx.fourier.FourierMeta attribute) copy_to (bijx.fourier.FourierMeta attribute) correlation_length() (in module bijx.lattice.scalar) count_primary (bijx.BinaryMask property) count_secondary (bijx.BinaryMask property) counts (bijx.BinaryMask property) cov (bijx.DiagonalNormal property) (bijx.MultivariateNormal property) covs (bijx.GaussianMixture property) create() (bijx.fourier.FourierMeta class method) create_eigenvalue_grid() (in module bijx.lie) crouch_grossmann() (in module bijx.cg) crouch_grossmann_step() (in module bijx.cg) curve_grad() (in module bijx.lie) cyclic_corr() (in module bijx.lattice.scalar) cyclic_corr_mat() (in module bijx.lattice.scalar) cyclic_tensor() (in module bijx.lattice.scalar)

D

data (bijx.fourier.FourierData attribute) default_wrap() (in module bijx) density() (bijx.Distribution method) DiagonalNormal (class in bijx)

DiffraxConfig (class in bijx) dim (bijx.DiagonalNormal property) (bijx.MultivariateNormal property) Distribution (class in bijx) dt (bijx.DiffraxConfig attribute)

E

effective_sample_size() (in module bijx) Embedding (class in bijx.nn.embeddings) EULER (in module bijx.cg) evaluate_density_on_eigenvalue_grid() (in module bijx.lie) event (bijx.DiffraxConfig attribute) event_axes (bijx.ArrayDistribution property) (bijx.ShapeInfo property) event_dim (bijx.ArrayDistribution property) (bijx.ShapeInfo attribute)

event_shape (bijx.BinaryMask attribute) (bijx.ShapeInfo attribute) event_size (bijx.ArrayDistribution property) (bijx.BinaryMask property) (bijx.ShapeInfo property) ExpandDims (class in bijx) Exponential (class in bijx)

F

fft_momenta() (in module bijx.fourier) FFTRep (class in bijx.fourier) flip() (bijx.BinaryMask method) flip_axis() (in module bijx.lattice.gauge) flip_lattice() (in module bijx.nn.conv) fold_kernel() (in module bijx.nn.conv) forward() (bijx.ApplyBijection method) (bijx.Bijection method) (bijx.BinaryMask method) (bijx.Chain method) (bijx.CondInverse method) (bijx.ContFlowCG method) (bijx.ContFlowDiffrax method) (bijx.ContFlowRK4 method) (bijx.Frozen method) (bijx.GeneralCouplingLayer method) (bijx.Inverse method) (bijx.MetaLayer method) (bijx.Partial method) (bijx.ScalarBijection method) (bijx.ScanChain method) (bijx.ToFourierData method) FourierData (class in bijx.fourier)

FourierFeatures (class in bijx.nn.features) FourierMeta (class in bijx.fourier) FreeTheoryScaling (class in bijx) from_ab() (bijx.cg.ButcherTableau class method) from_boolean_mask() (bijx.BinaryMask class method) from_indices() (bijx.BinaryMask class method) from_params() (bijx.ModuleReconstructor method) from_real() (bijx.fourier.FourierData class method) from_state() (bijx.ModuleReconstructor method) Frozen (class in bijx) FrozenFilter (in module bijx) fwd() (bijx.AffineLinear method) (bijx.BetaStretch method) (bijx.Exponential method) (bijx.GaussianCDF method) (bijx.Power method) (bijx.ScalarBijection method) (bijx.Scaling method) (bijx.Shift method) (bijx.Sigmoid method) (bijx.Sinh method) (bijx.SoftPlus method) (bijx.Tan method) (bijx.Tanh method)

G

gather_orbits() (in module bijx.nn.conv) GaussianCDF (class in bijx) GaussianMixture (class in bijx) GeneralCouplingLayer (class in bijx) get_batch_shape() (bijx.ArrayDistribution method) (bijx.Distribution method) (bijx.GaussianMixture method) (bijx.MixtureStack method)

get_complex_dtype() (bijx.fourier.FourierMeta method) given_cov() (bijx.MultivariateNormal class method) given_dim() (bijx.DiagonalNormal class method) (bijx.MultivariateNormal class method) given_variances() (bijx.DiagonalNormal class method) grad() (in module bijx.lie)

H

HaarDistribution (class in bijx.lie)

has_complex_params (bijx.ModuleReconstructor property) have_imag (bijx.fourier.FourierMeta property)

I

idc_copy_from (bijx.fourier.FourierMeta property) idc_copy_to (bijx.fourier.FourierMeta property) idc_have_imag (bijx.fourier.FourierMeta property) idc_rfft_independent (bijx.fourier.FourierMeta property) IMH (class in bijx) IMHInfo (class in bijx) IMHState (class in bijx) IndependentNormal (class in bijx)

IndependentUniform (class in bijx) indices() (bijx.BinaryMask method) init() (bijx.IMH method) Inverse (class in bijx) invert() (bijx.ApplyBijection method) (bijx.Bijection method) (bijx.CondInverse method) (bijx.Inverse method) is_accepted (bijx.IMHInfo attribute)

K

kernel (bijx.nn.conv.ConvSym property) kernel_d4() (in module bijx.nn.conv) kernel_equidist() (in module bijx.nn.conv) KernelFourier (class in bijx.nn.embeddings) KernelGauss (class in bijx.nn.embeddings)

KernelLin (class in bijx.nn.embeddings) KernelReduced (class in bijx.nn.embeddings) kinetic_term() (in module bijx.lattice.scalar) ks_full (bijx.fourier.FourierMeta attribute) ks_reduced (bijx.fourier.FourierMeta attribute)

L

load_shapes_magic() (in module bijx) log_density() (bijx.BufferedSampler method) (bijx.DiagonalNormal method) (bijx.Distribution method) (bijx.GaussianMixture method) (bijx.IndependentNormal method) (bijx.IndependentUniform method) (bijx.lie.HaarDistribution method) (bijx.MixtureStack method) (bijx.MultivariateNormal method) (bijx.Transformed method) log_jac() (bijx.AffineLinear method) (bijx.BetaStretch method) (bijx.Exponential method) (bijx.GaussianCDF method) (bijx.Power method) (bijx.ScalarBijection method) (bijx.Scaling method) (bijx.Shift method) (bijx.Sigmoid method) (bijx.Sinh method) (bijx.SoftPlus method) (bijx.Tan method) (bijx.Tanh method)

log_prob_proposal (bijx.IMHState attribute) log_prob_target (bijx.IMHState attribute)

M

made_jump (bijx.DiffraxConfig attribute) masks (bijx.BinaryMask attribute) max_steps (bijx.DiffraxConfig attribute) means (bijx.GaussianMixture property) meta (bijx.fourier.FourierData attribute) MetaLayer (class in bijx) mi (bijx.fourier.FourierMeta attribute) MixtureStack (class in bijx) MLP (class in bijx.nn.nets) module bijx bijx.cg bijx.fourier bijx.lattice bijx.lattice.gauge bijx.lattice.scalar bijx.lie bijx.nn.conv bijx.nn.embeddings bijx.nn.features bijx.nn.nets

ModuleReconstructor (class in bijx) MonotoneRQSpline (class in bijx) moving_average() (in module bijx) mr (bijx.fourier.FourierMeta attribute) MultivariateNormal (class in bijx)

N

noise_model() (in module bijx)

NonlinearFeatures (class in bijx.nn.features)

O

odeint_rk4() (in module bijx)

optional_override() (bijx.DiffraxConfig method)

P

param_count (bijx.MonotoneRQSpline property) param_splits (bijx.MonotoneRQSpline property) params (bijx.ModuleReconstructor property) params_array_splits (bijx.ModuleReconstructor property) params_dict (bijx.ModuleReconstructor property) params_dtypes (bijx.ModuleReconstructor property) params_shape_dict (bijx.ModuleReconstructor property) params_shapes (bijx.ModuleReconstructor property) params_sizes (bijx.ModuleReconstructor property) params_total_size (bijx.ModuleReconstructor property) Partial (class in bijx) path_div() (in module bijx.lie) path_grad() (in module bijx.lie)

path_grad2() (in module bijx.lie) periodic_gauge_lattice() (bijx.lie.HaarDistribution class method) phi4_term() (in module bijx.lattice.scalar) phi4_term_alt() (in module bijx.lattice.scalar) poly_term() (in module bijx.lattice.scalar) PolynomialFeatures (class in bijx.nn.features) position (bijx.IMHState attribute) PositionalEmbedding (class in bijx.nn.embeddings) Power (class in bijx) primary_indices (bijx.BinaryMask attribute) process_and_flatten() (bijx.ShapeInfo method) process_event() (bijx.ShapeInfo method) proposal (bijx.IMHInfo attribute) propose() (bijx.IMH method)

R

rational_quadratic_spline() (in module bijx) raw_value (bijx.Const attribute) rdof_to_complex() (bijx.fourier.FourierData static method) real_shape (bijx.fourier.FourierMeta property) real_space (bijx.fourier.FFTRep attribute) real_to_rfft() (bijx.fourier.FourierData static method) rep (bijx.fourier.FourierData attribute) replace() (bijx.cg.ButcherTableau method) (bijx.DiffraxConfig method) (bijx.fourier.FourierData method) (bijx.fourier.FourierMeta method) (bijx.IMHInfo method) (bijx.IMHState method) rescale_range() (in module bijx.nn.embeddings) Reshape (class in bijx) resize_kernel_weights() (in module bijx.nn.conv) ResNet (class in bijx.nn.nets) rev() (bijx.AffineLinear method) (bijx.BetaStretch method) (bijx.Exponential method) (bijx.GaussianCDF method) (bijx.Power method) (bijx.ScalarBijection method) (bijx.Scaling method) (bijx.Shift method) (bijx.Sigmoid method) (bijx.Sinh method) (bijx.SoftPlus method) (bijx.Tan method) (bijx.Tanh method)

reverse() (bijx.ApplyBijection method) (bijx.Bijection method) (bijx.BinaryMask method) (bijx.Chain method) (bijx.CondInverse method) (bijx.ContFlowCG method) (bijx.ContFlowDiffrax method) (bijx.ContFlowRK4 method) (bijx.Frozen method) (bijx.GeneralCouplingLayer method) (bijx.Inverse method) (bijx.MetaLayer method) (bijx.Partial method) (bijx.ScalarBijection method) (bijx.ScanChain method) (bijx.ToFourierData method) reverse_dkl() (in module bijx) rfft (bijx.fourier.FFTRep attribute) rfft_to_complex() (bijx.fourier.FourierData static method) rfft_to_real() (bijx.fourier.FourierData static method) roll_lattice() (in module bijx.lattice) rot_lattice_90() (in module bijx.nn.conv) rotate_lat() (in module bijx.lattice.gauge)

S

sample() (bijx.BufferedSampler method) (bijx.DiagonalNormal method) (bijx.Distribution method) (bijx.GaussianMixture method) (bijx.IndependentNormal method) (bijx.IndependentUniform method) (bijx.lie.HaarDistribution method) (bijx.MixtureStack method) (bijx.MultivariateNormal method) (bijx.Transformed method) sample_haar() (in module bijx.lie) saveat (bijx.DiffraxConfig attribute) scalar_prod() (in module bijx.lie) ScalarBijection (class in bijx) scale() (bijx.SpectrumScaling method) scales (bijx.DiagonalNormal property) scaling (bijx.FreeTheoryScaling property) (bijx.SpectrumScaling property) Scaling (class in bijx) ScanChain (class in bijx) secondary_indices (bijx.BinaryMask attribute) shape_info (bijx.fourier.FourierMeta attribute) ShapeInfo (class in bijx)

Shift (class in bijx) Sigmoid (class in bijx) Sinh (class in bijx) skew_traceless_cot() (in module bijx.lie) SoftPlus (class in bijx) solve() (bijx.DiffraxConfig method) solve_flow() (bijx.ContFlowCG method) (bijx.ContFlowDiffrax method) (bijx.ContFlowRK4 method) solver (bijx.DiffraxConfig attribute) solver_state (bijx.DiffraxConfig attribute) space_axes (bijx.ShapeInfo property) space_dim (bijx.ShapeInfo attribute) space_shape (bijx.ShapeInfo attribute) space_size (bijx.ShapeInfo property) spectrum_function() (bijx.FreeTheoryScaling method) SpectrumScaling (class in bijx) SqueezeDims (class in bijx) stack (bijx.MixtureStack property) stage_reduce() (in module bijx.cg) stages (bijx.cg.ButcherTableau attribute) step() (bijx.IMH method) stepsize_controller (bijx.DiffraxConfig attribute) swap_axes() (in module bijx.lattice.gauge)

T

t_end (bijx.DiffraxConfig attribute) t_start (bijx.DiffraxConfig attribute) Tan (class in bijx) Tanh (class in bijx) throw (bijx.DiffraxConfig attribute)

to() (bijx.fourier.FourierData method) ToFourierData (class in bijx) Transformed (class in bijx) transport() (in module bijx.cg) two_point() (in module bijx.lattice.scalar) two_point_central() (in module bijx.lattice.scalar)

U

unfold_kernel() (in module bijx.nn.conv) unique_idc (bijx.fourier.FourierMeta attribute)

unique_index_kernel() (in module bijx.nn.conv) unique_unfold (bijx.fourier.FourierMeta attribute)

V

value_grad_divergence() (in module bijx.lie)

variances (bijx.DiagonalNormal property)

W

weights (bijx.GaussianMixture property) weights_log_normalized (bijx.MixtureStack property)

weights_normalized (bijx.MixtureStack property) wilson_action() (in module bijx.lattice.gauge) wilson_log_prob() (in module bijx.lattice.gauge)