jax_rb.utils

Utils

various utils for the project

jax_rb.utils.utils.complement_basis_for_vector(xraw)

complement basis of xraw, a non zero vector. Assume x[0] !=0

jax_rb.utils.utils.make_complement_basis(x)

make complement basis of

jax_rb.utils.utils.generate_symmetric_tensor(key, k, m)

Generating symmetric tensor size k,m

jax_rb.utils.utils.esqrtm(x)

sqrtm by eigenvalue

jax_rb.utils.utils.unvec_skew(v)

unravel a n(n-1)//2 vector to anti hermitian matrix

jax_rb.utils.utils.jpolar(x)

jax polar decomposition

jax_rb.utils.utils.rand_positive_definite(key, n, bounds=None)

generate a positive definite matrix of size n

jax_rb.utils.utils.grand(key, dims)

generate a random array of shape dim using key