improve readthedocs behavior for jax.remat / jax.checkpoint#24064

mattjj wants to merge 1 commit intojax-ml:mainjax-ml/jax:mainfrom

mattjj:improve-checkpoint-doc-behaviormattjj/jax:improve-checkpoint-doc-behaviorCopy head branch name to clipboard