fixed save_exp clipping for compatibility with newest JAX versions#802

jnsbck merged 2 commits into

jaxleyverse:mainjaxleyverse/jaxley:mainfrom

lorenzosquadrani:fix-jax-clip-keywordlorenzosquadrani/jaxley:fix-jax-clip-keywordCopy head branch name to clipboard