Enable optional use of eig backends other than jax via the jeig package. jeig uses torch backend by default, which seems to offer consistent performance improvements.
It's a bit awkward to have a jax-based code depend on torch, so the dependency is optional and jeig is not used by default.
fmmax can be installed with jeig via pip install fmmax[jeig]
Enable optional use of eig backends other than jax via the jeig package. jeig uses torch backend by default, which seems to offer consistent performance improvements.
It's a bit awkward to have a jax-based code depend on torch, so the dependency is optional and jeig is not used by default.
fmmax can be installed with jeig via
pip install fmmax[jeig]