Justin Pan comments

Repositories
Issues
Comments

Results 1 comments of


                                            Justin Pan

saving with default/global mesh is broken

Hi there, thanks again for the report. Quick update: the issue should be fixed by https://github.com/google/orbax/commit/5a0bd479c39b86b31ef855e7770b461a18f3a3c8 As pointed out in the report, when a global mesh is active, `jax.lax.slice_in_dim` can...