fix array initialization problem when import package

[chaoming0625]

This pull request introduces significant changes to the braintaichi/_primitive/_ad_support.py file, focusing on enhancing the functionality for defining JVP (Jacobian-vector product) rules for JAX primitives. The most important changes include the addition of new imports, a redefined defjvp function, and the implementation of helper functions to support the new functionality.

Enhancements to JVP rule definitions:

• Added imports for functools, partial, jax, tree_util, and ad to support the new functionality.
• Redefined the defjvp function to handle primitives with multiple_results=True and added an example reference for usage.
• Implemented the _standard_jvp helper function to manage JVP rules for primitives with multiple results, ensuring compatibility with different versions of JAX.
• Added the _add_tangents helper function to combine tangents using tree_util.tree_map and ad.add_tangents.