Skip to content

fix serialization of partitioned states in optax (#852) #15

fix serialization of partitioned states in optax (#852)

fix serialization of partitioned states in optax (#852) #15