jax
030ee4a1 - Merge pull request #25070 from jax-ml:pjit-lin-rule

Commit

1 year ago

Merge pull request #25070 from jax-ml:pjit-lin-rule PiperOrigin-RevId: 699304829

References

#25070 - Add linearization rule for pjit_p

#170 - CI: 12/02/24 upstream sync

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading