cyjax.ml

Collection of machine learning functions and classes.

cholesky_decode(diag, upper)	Construct hermitian matrix from Cholesky decomposition.
cholesky_from_param(h_par)	Construct hermitian matrix from Cholesky decomposition parameters.
hermitian_param_init(key, basis_size[, ...])	Initialize parametrization to yield identity for Hermitian matrix.
variance_eta_loss(h, sample, algebraic_metric)	Compute variance-based eta loss.

BatchSampler(seed, variety, params_sampler)

HNetMLP(basis_size, layer_sizes[, ...])

Dense network for learning moduli dependence of the H matrix.