/svd2

A JAX/Numpy implementation of a batched 2x2 SVD

Primary LanguageJupyter Notebook

Stargazers