flax
Upgrade the `pjit` guide to `jax.jit`
#3016
Merged
Overview
Commits
2
Changes
View On GitHub
Commits
  • upgrade jit guide
    IvyZX committed 2 years ago
  • Update Scale up Flax Modules on multiple devices
    IvyZX committed 2 years ago
FAQ Terms Privacy Refunds Impressum
Loading