Backend JAX: learning rate decay with Optax#1992

lululxvi merged 3 commits intolululxvi:masterlululxvi/deepxde:masterfrom

vl-dud:jax-decay-schedulevl-dud/deepxde:jax-decay-scheduleCopy head branch name to clipboard