jax
ef84e9d3 - Rename `pbroadcast` to `pvary` and expose it as `jax.lax.pvary`.

Commit

1 year ago

Rename `pbroadcast` to `pvary` and expose it as `jax.lax.pvary`. PiperOrigin-RevId: 745342103

Author

yashk2810

Committer

charleshofer

Parents