aspcore.matrices_jax

Functions for common operations on matrices.

Some examples include constructing a block matrix, ensure positive definiteness, applying a function to individual blocks of a block matrix.

References

Functions

matmul_param(mat1, mat2)	Multiplies two parametrized block matrices without explicitly converting to full matrices.
regularize_matrix_with_condition_number(mat)	Adds a scaled identity matrix to the matrix in order to ensure a maximum condition number

Classes

partial

partial(func, *args, **keywords) - new function with partial application of the given arguments and keywords.