Closed Joshuaalbert closed 4 months ago
Shardable seeds jax.config.update('jax_threefry_partitionable', True)
64bit on for those parts that need it: but that's hard to control... But we must.
number of hosts set properly
Shardable seeds jax.config.update('jax_threefry_partitionable', True)
64bit on for those parts that need it: but that's hard to control... But we must.
number of hosts set properly