Solve the qwix bugs when config.per_device_batch_size * jax.local_device_count() < 1#237

entrpn merged 4 commits intomainAI-Hypercomputer/maxdiffusion:mainfrom

sanbao/bugsAI-Hypercomputer/maxdiffusion:sanbao/bugsCopy head branch name to clipboard