jax
e4992c3d - [state] Forward `jax.Ref.at` to the aval

Commit

30 days ago

[state] Forward `jax.Ref.at` to the aval PiperOrigin-RevId: 999309267

References

#41559 - [state] Forward `jax.Ref.at` to the aval

Author

superbobry

superbobry

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading