jax
351cd419
- [pallas:mgpu] Support `Element` block dim in pipeline block specs.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
121 days ago
[pallas:mgpu] Support `Element` block dim in pipeline block specs. PiperOrigin-RevId: 892888580
References
#36411 - [pallas:mgpu] Support `Element` block dim in pipeline block specs.
Author
chr1sj0nes
Committer
Google-ML-Automation
Parents
06669731
Loading
