Add `allow_ops_in_compiled_graph` to support torch.compile

arogozhnikov / einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

https://einops.rocks

MIT License

8.41k stars 350 forks source link

Closed arogozhnikov closed 1 year ago