flax
Use jax.api.device_put_sharded() in place of private JAX APIs.
#466
Merged
Overview
Commits
1
Changes
View On GitHub
FAQ Terms Privacy Refunds Impressum
Loading