/opt_einsum_fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

Primary LanguagePythonMIT LicenseMIT