Improve JAX compatibility with ics and bcs.#1492

tttc3 wants to merge 2 commits intolululxvi:masterlululxvi/deepxde:masterfrom

tttc3:jax-icbcstttc3/deepxde:jax-icbcsCopy head branch name to clipboard