Use jax.lax instead of jax.numpy in the backend

yhtang / FunFact

Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.

Other

57 stars 4 forks source link

Open yhtang opened 2 years ago