Registry of add_any and pad

[zgbkdlm]

Hi,

I have encountered the following warnings of unregistered functions. These functions add_any and pad seem to be JAX primitives and I don't find any intructions in README.md for how to register them. Do you have any example?

WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/jvp.py:493:15 (get_jvp_function.<locals>.merged_fwd)) - add_any not in registry. The following call might be slow as we will compute the full hessian.
WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/jvp.py:493:15 (get_jvp_function.<locals>.merged_fwd)) - add_any not in registry. The following call might be slow as we will compute the full hessian.
WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/utils.py:98:12 (flat_wrap.<locals>.new_fn)) - pad not in registry. The following call might be slow as we will compute the full hessian.
WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/utils.py:98:12 (flat_wrap.<locals>.new_fn)) - pad not in registry. The following call might be slow as we will compute the full hessian.
WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/ad.py:70:39 (jacrev.<locals>.jacfun.<locals>.flat_f)) - pad not in registry. The following call might be slow as we will compute the full hessian.
WARNING:[folx](/home/.venv/lib/python3.12/site-packages/folx/ad.py:70:39 (jacrev.<locals>.jacfun.<locals>.flat_f)) - pad not in registry. The following call might be slow as we will compute the full hessian.