Skip to content

Commit

Permalink
Update Scale up Flax Modules on multiple devices
Browse files Browse the repository at this point in the history
  • Loading branch information
8bitmp3 committed Jun 5, 2023
1 parent 34e18a5 commit 1db4c96
Show file tree
Hide file tree
Showing 2 changed files with 180 additions and 131 deletions.
Loading

0 comments on commit 1db4c96

Please sign in to comment.