jax
8bb19772 - Use shaped_abstractify to handle non-jax/duck types

Commit

40 days ago

Use shaped_abstractify to handle non-jax/duck types PiperOrigin-RevId: 903375845

References

#37045 - Use shaped_abstractify to handle non-jax/duck types

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading