flax
fd118b12 - use jax.shard_map
Commit
View On GitHub
Commit
82 days ago
use jax.shard_map PiperOrigin-RevId: 817884611
Author
Cristian Garcia
Committer
a-googler a-googler
Parents
5bf9b35f
FAQ Terms Privacy Refunds Impressum
Loading