Remove usages of orbax_utils.save_args_from_target, as this function does nothing (it used to control a checkpointing behavior that has since been optimized away).#4482
Open
copybara-service[bot] wants to merge 1 commit intomainfrom test_715575757
+2-4
