Linux-cpp-lisp / opt_einsum_fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Linux-cpp-lisp/opt_einsum_fx Stargazers