jax
c5dd0070 - [pallas] Allow `loop` functions to have a loop carry.

Commit

47 days ago

[pallas] Allow `loop` functions to have a loop carry. PiperOrigin-RevId: 899508681

References

Author

chr1sj0nes

Committer

Google-ML-Automation

Parents