v2.20.1

What's Changed

• Fix save_weights when saving weights in TF format with a remote path. by @copybara-service[bot] in #837
• Automated Code Change by @copybara-service[bot] in #839
• This change proactively relaxes floating point comparisons in unit tests to avoid breaking them when updating the Eigen library in cl/747497592. by @copybara-service[bot] in #841
• Internal change in a BUILD file. by @copybara-service[bot] in #840
• Update API goldens for Python 3.12. by @copybara-service[bot] in #842
• Adds support for Call-Context Arguments by @copybara-service[bot] in #843
• Disallow pickle loading in npz files. by @copybara-service[bot] in #849
• Initial implementation of PR-gain AUC metrics based on http://people.cs.bris.ac.uk/~flach/PRGcurves/PRcurves.pdf by @copybara-service[bot] in #845
• Reformat code that doesn't follow the standard. by @copybara-service[bot] in #859
• Add sample weight support; pass training=True in SAM's train_step to support layers like BatchNorm and Dropout. by @copybara-service[bot] in #858
• Add hour as a suffix to the wheel name. by @copybara-service[bot] in #860
• Add rng_type to BaseRandomLayer.get_config(). by @copybara-service[bot] in #852
• Patch version 0.20.1 by @sachinprasadhs in #861

New Contributors

• @sachinprasadhs made their first contribution in #861

Full Changelog: v2.19.0...v2.20.1