Fix surrogate gradient function and numpy 2.0 compatibility

[chaoming0625]

• Fix the batching rule bug of surrogate gradient functions
• Remove the following numpy APIs
  • add_docstring
  • add_newdoc
  • add_newdoc_ufunc
  • array2string
  • disp
  • issubclass_