fix: register_model in visualization_jax.py to actually exercise JIT path#37

Jammy2211 merged 1 commit into

mainPyAutoLabs/autogalaxy_workspace_test:mainfrom

feature/viz-jax-pytree-fixPyAutoLabs/autogalaxy_workspace_test:feature/viz-jax-pytree-fixCopy head branch name to clipboard