jax
87a4071b - [pallas] `pl.loop` now accepts `step=`

Commit

313 days ago

[pallas] `pl.loop` now accepts `step=` This makes the call site a bit more compact in cases where the loop needs a non-unit step. PiperOrigin-RevId: 785879334

References

#30274 - [pallas] `pl.loop` now accepts `step=`

Author

superbobry

Committer

Google-ML-Automation

Parents

eb94d52e

jax 87a4071b - [pallas] `pl.loop` now accepts `step=`

jax
87a4071b - [pallas] `pl.loop` now accepts `step=`