Skip to content
Snippets Groups Projects
Commit c87fdeb8 authored by Jakob Roth's avatar Jakob Roth
Browse files

optimize_kl: store mesh and pspec as NamedSharding

parent db221234
No related branches found
No related tags found
1 merge request!993Multi gpu
Pipeline #241425 passed
Stage: build_docker
Stage: test
Stage: demo_runs
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment