search

yhtang / FunFact

Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
Other
57 stars 4 forks source link

Additional arguments for `ff.use`. Closes #166. #193

Closed yhtang closed 2 years ago

yhtang commented 2 years ago

PyTorch:

ff.use('torch', device'cuda')

JAX

ff.use('jax', enable_x64=True)