jax
8bc11138
- Migrated ``pl.BlockSpec`` uses in JAX to the new argument order
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Migrated ``pl.BlockSpec`` uses in JAX to the new argument order See #22209. PiperOrigin-RevId: 648681171
References
#22221 - Migrated ``pl.BlockSpec`` uses in JAX to the new argument order
Author
superbobry
Committer
a-googler
Parents
de0fd722
Loading
