jax
dbc3bcd3 - Apply forwarding in pjit linearization rule to avoid intermediate copies.

Commit

331 days ago

Apply forwarding in pjit linearization rule to avoid intermediate copies.

References

#27735 - Apply forwarding in pjit linearization rule to avoid intermediate copies

Author

dfm

dfm

Committer

dfm

dfm

Parents

Loading