Closed SamDuffield closed 1 month ago
fmmax.utils.eig is jit-able and differentiable and allows us to remove the A_spd argument (which was difficult because for dense D then A_y is likely not symmetric positive definite anyway.
fmmax.utils.eig
A_spd
D
A_y
fmmax.utils.eigis jit-able and differentiable and allows us to remove theA_spdargument (which was difficult because for denseDthenA_yis likely not symmetric positive definite anyway.