Make jax optional in testing suite#380

ricardoV94 merged 1 commit intopymc-devs:mainpymc-devs/pymc-extras:mainfrom

ricardoV94:jax_optional_testingricardoV94/pymc-extras:jax_optional_testingCopy head branch name to clipboard