Remove dependence on old flax PRNG compat mode.

google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

Apache License 2.0

1.45k stars 202 forks source link

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Remove dependence on old flax PRNG compat mode.