[WIP] feat: adopt a more JAX-like API#211

beckermr wants to merge 18 commits into

mainGalSim-developers/JAX-GalSim:mainfrom

jax-api-changesGalSim-developers/JAX-GalSim:jax-api-changesCopy head branch name to clipboard