Skip to content

Added sharding parameter: fix for JAX 0.4.31 (#25) #12

Added sharding parameter: fix for JAX 0.4.31 (#25)

Added sharding parameter: fix for JAX 0.4.31 (#25) #12