jax
76d59380 - [pallas] Added `MemoryRef` and `run_scoped` to the API docs

Commit

1 year ago

[pallas] Added `MemoryRef` and `run_scoped` to the API docs PiperOrigin-RevId: 683349061

References

Author

superbobry

Committer

Google-ML-Automation

Parents