Modernize JAX compatibility and separate core dependencies from example extras#38

vimarsh6739 wants to merge 3 commits into

PredictiveIntelligenceLab:mainPredictiveIntelligenceLab/jaxpi:mainfrom

vimarsh6739:jax-compat-packagingvimarsh6739/jaxpi:jax-compat-packagingCopy head branch name to clipboard