Tweaks to DistributedEmbedding for JAX example.
#602
| Job | Run time |
|---|---|
| 24s | |
| 24s |
DistributedEmbedding for JAX example.
#602
| Job | Run time |
|---|---|
| 24s | |
| 24s |